TPUPartitionedOutputV2

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

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

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

รายการเอาต์พุตที่แบ่งพาร์ติชันซึ่งมีรูปร่างเหมือนกัน