CollectivePermute

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

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>
สร้าง (ขอบเขต ขอบเขต , อินพุต Operand <T>, Operand <Integer> sourceTargetPairs)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ CollectivePermute ใหม่
เอาท์พุต <T>
เอาท์พุท ()
อินพุตที่ดัดแปลง

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

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

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

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

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

CollectivePermute <T> แบบคงที่สาธารณะ สร้าง ( ขอบเขตขอบเขต อินพุต Operand <T> ตัวดำเนินการ <Integer> sourceTargetPairs)

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

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
ป้อนข้อมูล อินพุตท้องถิ่นที่จะสับเปลี่ยน ขณะนี้รองรับเฉพาะ float และ bfloat16 เท่านั้น
แหล่งที่มาTargetPairs เทนเซอร์ที่มีรูปร่าง [num_pairs, 2]
การส่งคืน
  • อินสแตนซ์ใหม่ของ CollectivePermute

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

อินพุตที่ดัดแปลง