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