XlaConcatND

เชื่อมต่อเทนเซอร์อินพุตเข้ากับทุกมิติ

op ที่ผสานการแบ่งเมตริกซ์อินพุตตามแอตทริบิวต์ num_splits ที่กำหนด ดึงช่องว่างเสริมออกหรือไม่ก็ได้ และส่งกลับเมตริกซ์ที่ผสานโดยไม่มีช่องว่างภายใน

Op นี้อาจสร้างขึ้นผ่านสะพาน TPU

ตัวอย่างเช่น ด้วยเทนเซอร์ "อินพุต":

[[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]]