TPUPartitionedOutputV2 คลาสสุดท้ายสาธารณะ
สหกรณ์ที่แยกเมตริกซ์เทนเซอร์ที่จะแบ่งส่วนโดย XLA ไปยังรายการพาร์ติชัน
เอาต์พุตนอกการคำนวณ XLA รองรับการแบ่งส่วน ND
วิธีการสาธารณะ
| คงที่ <T> TPUPartitionedOutputV2 <T> | สร้าง (ขอบเขต ขอบเขต , ตัวดำเนินการ <T> อินพุต, numSplits แบบยาว, รายการ <Long> partitionDims) วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ TPUPartitionedOutputV2 ใหม่ |
| ตัววนซ้ำ < ตัวดำเนินการ <T>> | ตัววนซ้ำ () |
| รายการ < เอาท์พุต <T>> | เอาท์พุท () รายการเอาต์พุตที่แบ่งพาร์ติชันซึ่งมีรูปร่างเหมือนกัน |
วิธีการสืบทอด
วิธีการสาธารณะ
สร้าง TPUPartitionedOutputV2 <T> แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , อินพุต Operand <T>, Long numSplits, รายการ <Long> partitionDims)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ TPUPartitionedOutputV2 ใหม่
พารามิเตอร์
| ขอบเขต | ขอบเขตปัจจุบัน |
|---|---|
| อินพุต | เทนเซอร์ซึ่งแสดงถึงรูปร่างเต็มของเทนเซอร์ที่แบ่งพาร์ติชัน |
| พาร์ติชั่น Dims | รายการจำนวนเต็มที่อธิบายวิธีการแบ่งพาร์ติชันแต่ละมิติ ความว่างเปล่าบ่งชี้ว่าอินพุตถูกจำลองแบบ |
การส่งคืน
- อินสแตนซ์ใหม่ของ TPUPartitionedOutputV2