XlaSplitND

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

แยกเทนเซอร์อินพุตออกจากทุกมิติ

op ที่แบ่งส่วนของเมตริกซ์อินพุตตามแอตทริบิวต์ num_splits ที่กำหนด เพิ่มการแบ่งส่วนหรือไม่ก็ได้ และส่งคืนส่วนต่างๆ ชิ้นจะถูกส่งกลับในลำดับแถวหลัก

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

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

[[0, 1, 2],
  [3, 4, 5],
  [6, 7, 8]]
 
`num_splits`:
[2, 2]
 
และ `paddings`:
[1, 1]
 
`เอาต์พุต` ที่คาดไว้คือ:
[[0, 1],
  [3, 4]]
 [[2, 0],
  [5, 0]]
 [[6, 7],
  [0, 0]]
 [[8, 0],
  [0, 0]]
 

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

ระดับ XlaSplitND.ตัวเลือก คุณลักษณะเพิ่มเติมสำหรับ XlaSplitND

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

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

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

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

การสร้าง XlaSplitND <T> แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , อินพุต Operand <T>, Long N, รายการ <Long> numSplits, ตัวเลือก... ตัวเลือก)

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

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

ตัววนซ้ำสาธารณะ < ตัวดำเนินการ <T>> ตัววนซ้ำ ()

รายการสาธารณะ < เอาท์พุต <T>> เอาท์พุต ()

การขยาย XlaSplitND.Options สาธารณะแบบคงที่ (รายการ <Long> การขยาย)

พารามิเตอร์
ช่องว่างภายใน รายการเสริมของช่องว่างด้านขวาต่อมิติของเทนเซอร์อินพุตที่จะใช้ก่อนการแยก ซึ่งสามารถใช้เพื่อทำให้มิติแบ่งเท่าๆ กัน