TPUPartitionedOutput

TPUPartitionedOutput คลาสสุดท้ายสาธารณะ

สหกรณ์ที่แยกเมตริกซ์เทนเซอร์ที่จะแบ่งส่วนโดย XLA ไปยังรายการพาร์ติชัน

เอาต์พุตนอกการคำนวณ XLA

คลาสที่ซ้อนกัน

ระดับ TPUPartitionedOutput.Options แอตทริบิวต์ทางเลือกสำหรับ TPUPartitionedOutput

วิธีการสาธารณะ

คงที่ <T> TPUPartitionedOutput <T>
สร้าง (ขอบเขต ขอบเขต , ตัวดำเนินการ <T> อินพุต, numSplits แบบยาว, ตัวเลือก... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ TPUPartitionedOutput ใหม่
ตัววนซ้ำ < ตัวดำเนินการ <T>>
รายการ < เอาท์พุต <T>>
เอาท์พุท ()
รายการอินพุตที่แบ่งพาร์ติชันซึ่งจะต้องมีรูปร่างเหมือนกัน
TPUPartitionedOutput.Options แบบคงที่
partitionDim (พาร์ติชันยาวDim)

วิธีการสืบทอด

วิธีการสาธารณะ

สร้าง TPUPartitionedOutput <T> แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , อินพุต Operand <T>, Long numSplits, ตัวเลือก... ตัวเลือก)

วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ TPUPartitionedOutput ใหม่

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
อินพุต เทนเซอร์ซึ่งแสดงถึงรูปร่างเต็มของเทนเซอร์ที่แบ่งพาร์ติชัน
ตัวเลือก มีค่าแอตทริบิวต์ทางเลือก
การส่งคืน
  • อินสแตนซ์ใหม่ของ TPUPartitionedOutput

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

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

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

สาธารณะ TPUPartitionedOutput.Options partitionDim แบบคงที่ (Long partitionDim)

พารามิเตอร์
พาร์ทิชันDim จำนวนเต็มจะอธิบายว่ามิติใดถูกแบ่งพาร์ติชัน