AssignVariableXlaConcatND

کلاس نهایی عمومی AssignVariableXlaConcatND

تانسور ورودی Concat در تمام ابعاد.

عملیاتی که ادغام می شود، تانسور ورودی را بر اساس ویژگی num_splits داده شده برش می دهد، padding ها را به صورت اختیاری جدا می کند و تانسور ادغام شده را بدون بالشتک در متغیر منبع می نویسد.

این عملیات ممکن است از طریق پل TPU ایجاد شود.

برای مثال، با تانسور «ورودی»:

[[0, 1],
  [4, 5]]
 [[2, 3],
  [6, 7]]
 [[8, 9],
  [12, 13]]
 [[10, 11],
  [14, 15]]
 
«num_splits»:
[2, 2]
 
و «paddings»:
[1, 1]
 
، «خروجی» مورد انتظار عبارت است از:
[[0, 1, 2],
  [4, 5, 6],
  [8, 9, 10]]
 

کلاس های تو در تو

کلاس AssignVariableXlaConcatND.Options ویژگی های اختیاری برای AssignVariableXlaConcatND

روش های عمومی

استاتیک <T> AssignVariableXlaConcatND
ایجاد ( دامنه دامنه ، منبع عملوند <?>، ورودی های تکرارپذیر< Operand <T>>، لیست<Long> numConcats، گزینه ها... گزینه ها)
روش کارخانه برای ایجاد کلاسی که یک عملیات AssignVariableXlaConcatND جدید را بسته بندی می کند.
استاتیک AssignVariableXlaConcatND.Options
بالشتک ها (List<Long> paddings)

روش های ارثی

روش های عمومی

عمومی استاتیک AssignVariableXlaConcatND ایجاد ( دامنه دامنه ، منبع عملوند <?>، ورودی های تکرارپذیر< Operand <T>>، لیست <طولانی> numConcats، گزینه ها... گزینه ها)

روش کارخانه برای ایجاد کلاسی که یک عملیات AssignVariableXlaConcatND جدید را بسته بندی می کند.

مولفه های
محدوده محدوده فعلی
منبع متغیر منبع برای تانسورهای ورودی به هم پیوسته در تمام ابعاد. } in_arg { name: "inputs" توضیحات: <
numConcats تعداد روش های ادغام در هر بعد.
گزینه ها مقادیر ویژگی های اختیاری را حمل می کند
برمی گرداند
  • یک نمونه جدید از AssignVariableXlaConcatND

استاتیک عمومی AssignVariableXlaConcatND.Options paddings (List<Long> paddings)

مولفه های
بالشتک ها فهرست اختیاری از لایه های سمت راست در هر بعد برای حذف از تانسور ادغام شده نهایی. این بالشتک ها نباید از اندازه بعد نتیجه ادغام شده قبل از برداشتن لایه ها بیشتر شود.