SparseSegmentMeanGradV2

lớp cuối cùng công khai SparseSegmentMeanGradV2

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>
tạo ( Phạm vi phạm vi, Toán hạng <T> grad, Chỉ số toán hạng <U>, Toán tử <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.
Đầu ra <T>
đầu ra ()
Đầu ra <U>
được sắp xếpUniqueIndices ()

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

Đầu ra công khai đầu ra <T> ()

Đầu ra công khai <U> được sắp xếpUniqueIndices ()