Tính toán độ dốc cho SparseSegmentMean.
Trả về "đầu ra" tensor có hình dạng giống như grad, ngoại trừ thứ nguyên 0 có giá trị là số chỉ mục duy nhất trong "chỉ số". Cũng trả về vectơ "sort_unique_indices" chứa các chỉ mục tương ứng từ "chỉ số".
Phương pháp công khai
tĩnh <T mở rộng Số, U mở rộng Số, V mở rộng Số> SparseSegmentMeanGradV2 <T, U> | |
Đầu ra <T> | đầu ra () |
Đầu ra <U> |
Phương pháp kế thừa
Phương pháp công khai
public static SparseSegmentMeanGradV2 <T, U> tạo ( Phạm vi phạm vi, Toán hạng <T> grad, Chỉ số toán hạng <U>, Toán hạng <V> SegmentIds, Toán hạng <Integer> dày đặcOutputDim0)
Phương thức gốc để tạo một lớp bao bọc một thao tác SparseSegmentMeanGradV2 mới.
Thông số
phạm vi | phạm vi hiện tại |
---|---|
tốt nghiệp | gradient được truyền tới op SparseSegmentMean. |
chỉ số | các chỉ mục được chuyển tới op SparseSegmentMean tương ứng. |
id phân đoạn | Seg_ids được chuyển tới op SparseSegmentMean tương ứng. |
dày đặcĐầu raDim0 | thứ nguyên 0 của "dữ liệu" được chuyển tới SparseSegmentMean op. |
Trả lại
- một phiên bản mới của SparseSegmentMeanGradV2