คลาสสุดท้ายสาธารณะ OutfeedDequeueTupleV2
ดึงค่าหลายค่าจากเอาท์พุตการคำนวณ ลำดับอุปกรณ์คือเทนเซอร์ที่ให้เอาท์พุตแบบไดนามิก
การดำเนินการนี้จะบล็อกอย่างไม่มีกำหนดจนกว่าข้อมูลจะพร้อมใช้งาน เอาต์พุต `i` สอดคล้องกับองค์ประกอบทูเปิล XLA `i`
วิธีการสาธารณะ
OutfeedDequeueTupleV2 แบบคงที่ | สร้าง ( ขอบเขต ขอบเขต ตัวดำเนินการ <Integer> deviceOrdinal, รายการ<Class<?>> dtypes, รายการ< รูปร่าง > รูปร่าง) วิธีการจากโรงงานเพื่อสร้างคลาสที่ห่อการดำเนินการ OutfeedDequeueTupleV2 ใหม่ |
ตัววนซ้ำ < ตัวดำเนินการ <วัตถุ>> | ตัววนซ้ำ () |
รายการ < เอาท์พุต <?>> | เอาท์พุท () รายการเทนเซอร์ที่จะอ่านจากเอาท์พุต |
วิธีการสืบทอด
วิธีการสาธารณะ
สร้าง OutfeedDequeueTupleV2 แบบคงที่สาธารณะ (ขอบเขต ขอบเขต , ตัวดำเนินการ <Integer> deviceOrdinal, รายการ <Class<?>> dtypes, รายการ < รูปร่าง > รูปร่าง)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ห่อการดำเนินการ OutfeedDequeueTupleV2 ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
อุปกรณ์ลำดับ | เทนเซอร์สเกลาร์ int ซึ่งเป็นตัวแทนของอุปกรณ์ TPU ที่จะใช้ นี่ควรเป็น -1 เมื่อ Op ทำงานบนอุปกรณ์ TPU และ >= 0 เมื่อ Op ทำงานบนอุปกรณ์ CPU |
ประเภท | ประเภทองค์ประกอบของแต่ละองค์ประกอบใน "เอาต์พุต" |
รูปร่าง | รูปร่างของเทนเซอร์แต่ละตัวใน "เอาท์พุต" |
การส่งคืน
- อินสแตนซ์ใหม่ของ OutfeedDequeueTupleV2