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