XlaConcatND

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

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

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

به عنوان مثال، با `تانسور input`:

[[0, 1],
  [4, 5]]
 [[2, 3],
  [6, 7]]
 [[8, 9],
  [12, 13]]
 [[10, 11],
  [14, 15]]
 

[2, 2]
 

[1, 1]
 

[[0, 1, 2],
  [4, 5, 6],
  [8, 9, 10]]