XlaSplitND คลาสสุดท้ายสาธารณะ
แยกเทนเซอร์อินพุตออกจากทุกมิติ
op ที่แบ่งส่วนของเมตริกซ์อินพุตตามแอตทริบิวต์ num_splits ที่กำหนด เพิ่มการแบ่งส่วนหรือไม่ก็ได้ และส่งคืนส่วนต่างๆ ชิ้นจะถูกส่งกลับในลำดับแถวหลัก
Op นี้อาจสร้างขึ้นผ่านสะพาน TPU
ตัวอย่างเช่น ด้วยเทนเซอร์ "อินพุต":
[[0, 1, 2],
[3, 4, 5],
[6, 7, 8]]
[2, 2]
[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
การขยาย XlaSplitND.Options สาธารณะแบบคงที่ (รายการ <Long> การขยาย)
พารามิเตอร์
ช่องว่างภายใน | รายการเสริมของช่องว่างด้านขวาต่อมิติของเทนเซอร์อินพุตที่จะใช้ก่อนการแยก สามารถใช้เพื่อทำให้มิติแบ่งเท่าๆ กัน |
---|