คำเตือน: API นี้เลิกใช้งานแล้วและจะถูกลบออกใน TensorFlow เวอร์ชันในอนาคตหลังจาก การแทนที่ เสถียรแล้ว

CollectivePermute

ประชาชน CollectivePermute ชั้นสุดท้าย

การดำเนินการเปลี่ยนเทนเซอร์ในอินสแตนซ์ TPU ที่จำลองแบบ

แต่ละอินสแตนซ์จะป้อนข้อมูลของตนเอง

ตัวอย่างเช่น สมมติว่ามีอินสแตนซ์ TPU 4 รายการ: `[A, B, C, D]` ผ่าน source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` รับผลลัพธ์: `[D, A, B, C]`

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

เอาท์พุท <T>
asOutput ()
ส่งกลับค่าแฮนเดิลเชิงสัญลักษณ์ของเทนเซอร์
คง <T> CollectivePermute <T>
สร้าง ( ขอบเขต ขอบเขต Operand <T> การป้อนข้อมูล ตัวถูกดำเนินการ <Integer> sourceTargetPairs)
วิธีการ Factory เพื่อสร้างคลาสที่ปิดการดำเนินการ CollectivePermute ใหม่
เอาท์พุท <T>
เอาท์พุท ()
อินพุตที่เปลี่ยนแปลง

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

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

ประชาชน เอาท์พุท <T> asOutput ()

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

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

สาธารณะคง CollectivePermute <T> สร้าง ( ขอบเขต ขอบเขต Operand <T> การป้อนข้อมูล ตัวถูกดำเนินการ <Integer> sourceTargetPairs)

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

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

ประชาชน เอาท์พุท <T> เอาท์พุท ()

อินพุตที่เปลี่ยนแปลง