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