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]`

ค่าคงที่

สตริง OP_NAME ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow

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

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

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

ค่าคงที่

สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME

ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow

ค่าคงที่: "CollectivePermute"

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

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

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

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

CollectivePermute <T> สาธารณะแบบคงที่ สร้าง (ขอบเขต ขอบเขต , อินพุต Operand <T>, Operand < TInt32 > sourceTargetPairs)

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

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

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

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