SparseSegmentSumGradV2

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

Tính toán độ dốc cho SparseSegmentSum.

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

static <T mở rộng Số, U mở rộng Số, V mở rộng Số> SparseSegmentSumGradV2 <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 SparseSegmentSumGradV2 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 SparseSegmentSumGradV2 <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 SparseSegmentSumGradV2 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 SparseSegmentSum op.
chỉ số các chỉ mục được chuyển tới op SparseSegmentSum tương ứng.
id phân đoạn Seg_ids được chuyển tới op SparseSegmentSum tương ứng.
dày đặcĐầu raDim0 thứ nguyên 0 của "dữ liệu" được chuyển tới SparseSegmentSum op.
Trả lại
  • một phiên bản mới của SparseSegmentSumGradV2

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

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