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> | |
| เอาท์พุต <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