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