TPUOrdinalSelector

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

ตัวเลือกแกน TPU Op

Op นี้จะสร้างชุดของแกน TPU (สำหรับการอุ่นเครื่อง) หรือแกน TPU เดี่ยว (สำหรับการอนุมานปกติ) เพื่อรันโปรแกรม TPU TPUPartitionedCall ใช้เอาต์พุต

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

เอาต์พุต <จำนวนเต็ม>
เป็นเอาท์พุต ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
TPUOrdinalSelector แบบคงที่
สร้าง ( ขอบเขต ขอบเขต)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ TPUOrdinalSelector ใหม่
เอาต์พุต <จำนวนเต็ม>
อุปกรณ์ลำดับ ()
เวกเตอร์ 1 แกน TPU ขึ้นไป

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

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

เอาท์พุท สาธารณะ <จำนวนเต็ม> asOutput ()

ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์

อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต

สร้าง TPUOrdinalSelector สาธารณะแบบคงที่ (ขอบเขต ขอบเขต )

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

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
การส่งคืน
  • อินสแตนซ์ใหม่ของ TPUOrdinalSelector

สาธารณะ เอาท์พุท <จำนวนเต็ม> deviceOrdinals ()

เวกเตอร์ 1 แกน TPU ขึ้นไป