AssignVariableXlaConcatND

คลาสสุดท้ายสาธารณะ AssignVariableXlaConcatND

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

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]
 
และ `paddings`:
[1, 1]
 
`เอาต์พุต` ที่คาดไว้คือ:
[[0, 1, 2],
  [4, 5, 6],
  [8, 9, 10]]
 

คลาสที่ซ้อนกัน

ระดับ AssignVariableXlaConcatND.Options แอ็ตทริบิวต์ทางเลือกสำหรับ AssignVariableXlaConcatND

วิธีการสาธารณะ

คงที่ <T> AssignVariableXlaConcatND
สร้าง ( ขอบเขต ขอบเขต , ตัวดำเนินการ <?> ทรัพยากร, อินพุต < ตัวดำเนินการ <T>> ที่สามารถทำซ้ำได้, รายการ <Long> numConcats, ตัวเลือก... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ AssignVariableXlaConcatND ใหม่
AssignVariableXlaConcatND.Options แบบคงที่
ช่องว่างภายใน (รายการ <Long> ช่องว่างภายใน)

วิธีการสืบทอด

วิธีการสาธารณะ

AssignVariableXlaConcatND แบบคงที่สาธารณะ สร้าง (ขอบเขต ขอบเขต , ทรัพยากร ตัวดำเนินการ <?>, อินพุต < ตัวดำเนินการ <T>> ที่สามารถทำซ้ำได้, รายการ <Long> numConcats, ตัวเลือก... ตัวเลือก)

วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ AssignVariableXlaConcatND ใหม่

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
ทรัพยากร ตัวแปรทรัพยากรสำหรับเทนเซอร์อินพุตที่ต่อกันในทุกมิติ } in_arg { ชื่อ: "อินพุต" คำอธิบาย: <
numConcats จำนวนวิธีในการผสานต่อมิติ
ตัวเลือก มีค่าแอตทริบิวต์ทางเลือก
การส่งคืน
  • อินสแตนซ์ใหม่ของ AssignVariableXlaConcatND

สาธารณะคง AssignVariableXlaConcatND.Options paddings (รายการ <Long> paddings)

พารามิเตอร์
ช่องว่างภายใน รายการเสริมของช่องว่างด้านขวาต่อมิติที่จะแยกออกจากเทนเซอร์ที่รวมเข้าด้วยกันขั้นสุดท้าย ช่องว่างภายในเหล่านี้ต้องไม่เกินขนาดขนาดของผลลัพธ์ที่รวมเข้าด้วยกัน ก่อนที่จะลอกช่องว่างภายใน