Concats เทนเซอร์อินพุตในทุกมิติ
op ที่ผสานการแบ่งส่วนของอินพุตเทนเซอร์ตามแอตทริบิวต์ num_splits ที่กำหนด แยกช่องว่างภายในตัวเลือก และเขียนเทนเซอร์ที่ผสานโดยไม่มีช่องว่างภายในไปยังตัวแปรทรัพยากร
op นี้อาจสร้างขึ้นผ่านบริดจ์ TPU
ยกตัวอย่างเช่นด้วย `เมตริกซ์ input`:
[[0, 1],
[4, 5]]
[[2, 3],
[6, 7]]
[[8, 9],
[12, 13]]
[[10, 11],
[14, 15]]
` num_splits`: [2, 2]
และ `paddings`: [1, 1]
ที่คาดว่า` outputs` คือ [[0, 1, 2],
[4, 5, 6],
[8, 9, 10]]
คลาสที่ซ้อนกัน
ระดับ | AssignVariableXlaConcatND.ตัวเลือก | คุณลักษณะที่จำเป็นสำหรับการ AssignVariableXlaConcatND |
วิธีการสาธารณะ
คง <T> AssignVariableXlaConcatND | สร้าง ( ขอบเขต ขอบเขต Operand <?> ทรัพยากร Iterable < Operand <T >> ปัจจัยการผลิตรายการ <ยาว> numConcats, ตัวเลือก ... ตัวเลือก) วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ AssignVariableXlaConcatND ใหม่ |
คง AssignVariableXlaConcatND.Options | paddings (รายการ <ยาว> paddings) |
วิธีการสืบทอด
วิธีการสาธารณะ
สาธารณะคง AssignVariableXlaConcatND สร้าง ( ขอบเขต ขอบเขต Operand <?> ทรัพยากร Iterable < Operand <T >> ปัจจัยการผลิตรายการ <ยาว> numConcats, ตัวเลือก ... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ AssignVariableXlaConcatND ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
ทรัพยากร | ตัวแปรทรัพยากรสำหรับเทนเซอร์อินพุตที่ต่อกันในทุกมิติ } in_arg { ชื่อ: "อินพุต" คำอธิบาย: < |
numConcats | จำนวนวิธีในการผสานต่อมิติ |
ตัวเลือก | ดำเนินการค่าแอตทริบิวต์ทางเลือก |
คืนสินค้า
- อินสแตนซ์ใหม่ของ AssignVariableXlaConcatND
สาธารณะคง AssignVariableXlaConcatND.Options paddings (รายการ <ยาว> paddings)
พารามิเตอร์
paddings | รายการทางเลือกของช่องว่างภายในที่เหมาะสมต่อมิติเพื่อดึงออกจากเทนเซอร์ที่ผสานสุดท้าย ช่องว่างภายในเหล่านี้ต้องไม่เกินขนาดมิติของผลลัพธ์ที่ผสานก่อนที่จะลอกส่วนเสริม |
---|