Op เพื่อเปลี่ยนเทนเซอร์ข้ามอินสแตนซ์ TPU ที่จำลองแบบ
แต่ละอินสแตนซ์จะมีอินพุตของตัวเอง
ตัวอย่างเช่น สมมติว่ามีอินสแตนซ์ TPU 4 รายการ: `[A, B, C, D]` การส่งผ่าน source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` จะได้ผลลัพธ์: `[D, A, B, C]`
วิธีการสาธารณะ
เอาท์พุต <T> | เป็นเอาท์พุต () ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์ |
คง <T> CollectivePermute <T> | |
เอาท์พุต <T> | เอาท์พุท () อินพุตที่ดัดแปลง |
วิธีการสืบทอด
วิธีการสาธารณะ
เอาท์ พุท สาธารณะ <T> asOutput ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต
CollectivePermute <T> แบบคงที่สาธารณะ สร้าง ( ขอบเขตขอบเขต อินพุต Operand <T> ตัวดำเนินการ <Integer> sourceTargetPairs)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ CollectivePermute ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
ป้อนข้อมูล | อินพุตท้องถิ่นที่จะสับเปลี่ยน ขณะนี้รองรับเฉพาะ float และ bfloat16 เท่านั้น |
แหล่งที่มาTargetPairs | เทนเซอร์ที่มีรูปร่าง [num_pairs, 2] |
การส่งคืน
- อินสแตนซ์ใหม่ของ CollectivePermute