คลาสสุดท้ายสาธารณะ SparseSegmentSumGradV2
คำนวณการไล่ระดับสีสำหรับ SparseSegmentSum
ส่งกลับเทนเซอร์ "เอาต์พุต" ที่มีรูปร่างเหมือนกับ grad ยกเว้นมิติ 0 ซึ่งค่าคือจำนวนดัชนีที่ไม่ซ้ำกันใน "ดัชนี" นอกจากนี้ยังส่งคืนเวกเตอร์ "sorted_unique_indices" ที่มีดัชนีที่เกี่ยวข้องจาก "indices"
วิธีการสาธารณะ
คงที่ <T ขยายหมายเลข, U ขยายหมายเลข, V ขยายหมายเลข> SparseSegmentSumGradV2 <T, U> | สร้าง (ขอบเขต ขอบเขต , ตัวดำเนินการ <T> ผู้สำเร็จการศึกษา, ตัวดำเนินการ <U> ดัชนี, ตัวดำเนินการ <V> SegmentIds, ตัวดำเนินการ <Integer> หนาแน่นOutputDim0) วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ SparseSegmentSumGradV2 ใหม่ |
เอาท์พุต <T> | เอาท์พุท () |
เอาท์พุต <U> |
วิธีการสืบทอด
วิธีการสาธารณะ
สาธารณะ SparseSegmentSumGradV2 <T, U> สร้าง แบบคงที่ (ขอบเขต ขอบเขต , ตัวดำเนินการ <T> grad, ตัวดำเนินการ <U> ดัชนี, ตัวดำเนินการ <V> SegmentIds, ตัวดำเนินการ <Integer>หนาแน่นOutputDim0)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ SparseSegmentSumGradV2 ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
ผู้สำเร็จการศึกษา | การไล่ระดับสีแพร่กระจายไปยัง SparseSegmentSum op |
ดัชนี | ดัชนีที่ส่งผ่านไปยัง SparseSegmentSum op ที่เกี่ยวข้อง |
รหัสส่วน | Segment_ids ส่งผ่านไปยัง SparseSegmentSum op ที่เกี่ยวข้อง |
หนาแน่นOutputDim0 | มิติข้อมูล 0 ของ "ข้อมูล" ส่งผ่านไปยัง SparseSegmentSum op |
การส่งคืน
- อินสแตนซ์ใหม่ของ SparseSegmentSumGradV2