ประชาชน TPUPartitionedOutput ชั้นสุดท้าย
op ที่ demultiplexes เทนเซอร์ที่จะแยกส่วนโดย XLA ไปยังรายการของพาร์ติชั่น
เอาต์พุตนอกการคำนวณ XLA
คลาสที่ซ้อนกัน
ระดับ | TPUPartitionedOutput.ตัวเลือก | คุณลักษณะที่จำเป็นสำหรับการ TPUPartitionedOutput |
วิธีการสาธารณะ
คง <T> TPUPartitionedOutput <T> | สร้าง ( ขอบเขต ขอบเขต Operand <T> ปัจจัยการผลิต numSplits ยาว ตัวเลือก ... ตัวเลือก) วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ TPUPartitionedOutput ใหม่ |
Iterator < Operand <T >> | iterator () |
รายการ < เอาท์พุท <T >> | เอาท์พุท () รายการอินพุตที่แบ่งพาร์ติชันซึ่งต้องมีรูปร่างเหมือนกัน |
คง TPUPartitionedOutput.Options | partitionDim (Long partitionDim) |
วิธีการสืบทอด
วิธีการสาธารณะ
สาธารณะคง TPUPartitionedOutput <T> สร้าง ( ขอบเขต ขอบเขต Operand <T> ปัจจัยการผลิต numSplits ยาว ตัวเลือก ... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ TPUPartitionedOutput ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
อินพุต | เทนเซอร์ซึ่งแสดงถึงรูปร่างที่สมบูรณ์ของเทนเซอร์แบบแบ่งส่วน |
ตัวเลือก | ดำเนินการค่าแอตทริบิวต์ทางเลือก |
คืนสินค้า
- อินสแตนซ์ใหม่ของ TPUPartitionedOutput
สาธารณะคง TPUPartitionedOutput.Options partitionDim (Long partitionDim)
พารามิเตอร์
พาร์ทิชันDim | จำนวนเต็มอธิบายว่ามิติใดถูกแบ่งพาร์ติชัน |
---|