SparseSegmentSumGradV2

kelas akhir publik SparseSegmentSumGradV2

Menghitung gradien untuk SparseSegmentSum.

Mengembalikan "output" tensor dengan bentuk yang sama dengan grad, kecuali dimensi 0 yang nilainya adalah jumlah indeks unik dalam "indeks". Juga mengembalikan vektor "sorted_unique_indices" yang berisi indeks terkait dari "indeks".

Metode Publik

statis <T memperluas Nomor, U memperluas Nomor, V memperluas Nomor> SparseSegmentSumGradV2 <T, U>
buat ( Lingkup lingkup, Operand <T> grad, Operand <U> indeks, Operand <V> segmentIds, Operand <Integer> solidOutputDim0)
Metode pabrik untuk membuat kelas yang membungkus operasi SparseSegmentSumGradV2 baru.
Keluaran <T>
Keluaran <U>

Metode Warisan

Metode Publik

public static SparseSegmentSumGradV2 <T, U> buat ( Lingkup lingkup, Operan <T> grad, Indeks Operan <U>, Operan <V> segmentIds, Operan <Bilangan Bulat> padatOutputDim0)

Metode pabrik untuk membuat kelas yang membungkus operasi SparseSegmentSumGradV2 baru.

Parameter
cakupan ruang lingkup saat ini
lulusan gradien disebarkan ke operasi SparseSegmentSum.
indeks indeks diteruskan ke operasi SparseSegmentSum yang sesuai.
segmenId segment_ids diteruskan ke operasi SparseSegmentSum yang sesuai.
padatOutputDim0 dimensi 0 dari "data" diteruskan ke operasi SparseSegmentSum.
Kembali
  • contoh baru dari SparseSegmentSumGradV2

Keluaran publik <T> keluaran ()

Output publik <U> diurutkanUniqueIndices ()