SparseSegmentSqrtNGradV2

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

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

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

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

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