ประชาชน XlaSplitND ชั้นสุดท้าย
แยกเมตริกซ์อินพุตในทุกมิติ
op ที่แบ่งเมตริกซ์อินพุตตามแอตทริบิวต์ num_splits ที่กำหนด แบ่งแผ่นตัวเลือก และส่งคืนสไลซ์ ชิ้นจะถูกส่งคืนในลำดับหลักแถว
op นี้อาจสร้างขึ้นผ่านบริดจ์ TPU
ยกตัวอย่างเช่นด้วย `เมตริกซ์ input`:
[[0, 1, 2],
[3, 4, 5],
[6, 7, 8]]
` num_splits`: [2, 2]
และ `paddings`: [1, 1]
ที่คาดว่า` outputs` คือ [[0, 1],
[3, 4]]
[[2, 0],
[5, 0]]
[[6, 7],
[0, 0]]
[[8, 0],
[0, 0]]
คลาสที่ซ้อนกัน
ระดับ | XlaSplitND.ตัวเลือก | คุณลักษณะที่จำเป็นสำหรับการ XlaSplitND |
วิธีการสาธารณะ
คง <T> XlaSplitND <T> | สร้าง ( ขอบเขต ขอบเขต Operand <T> อินพุตยาว N, รายการ <ยาว> numSplits, ตัวเลือก ... ตัวเลือก) วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ XlaSplitND ใหม่ |
Iterator < Operand <T >> | iterator () |
รายการ < เอาท์พุท <T >> | เอาท์พุท () |
คง XlaSplitND.Options | paddings (รายการ <ยาว> paddings) |
วิธีการสืบทอด
วิธีการสาธารณะ
สาธารณะคง XlaSplitND <T> สร้าง ( ขอบเขต ขอบเขต Operand <T> อินพุตยาว N, รายการ <ยาว> numSplits, ตัวเลือก ... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ XlaSplitND ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
ป้อนข้อมูล | อินพุตเทนเซอร์เพื่อแบ่งตามมิติข้อมูลทั้งหมด } out_arg { ชื่อ: "เอาต์พุต" คำอธิบาย: < |
numSplits | จำนวนวิธีแยกต่อมิติ ขนาดรูปร่างต้องแบ่งเท่าๆ กัน |
ตัวเลือก | ดำเนินการค่าแอตทริบิวต์ทางเลือก |
คืนสินค้า
- อินสแตนซ์ใหม่ของ XlaSplitND
สาธารณะคง XlaSplitND.Options paddings (รายการ <ยาว> paddings)
พารามิเตอร์
paddings | รายการเสริมของช่องว่างด้านขวาต่อมิติของเทนเซอร์อินพุตที่จะใช้ก่อนแยก สามารถใช้เพื่อทำให้มิติสามารถหารได้เท่าๆ กัน |
---|