SparseSegmentMeanGradV2

kelas akhir publik SparseSegmentMeanGradV2

Menghitung gradien untuk SparseSegmentMean.

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> SparseSegmentMeanGradV2 <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 SparseSegmentMeanGradV2 baru.
Keluaran <T>
Keluaran <U>

Metode Warisan

Metode Publik

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

Metode pabrik untuk membuat kelas yang membungkus operasi SparseSegmentMeanGradV2 baru.

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

Keluaran publik <T> keluaran ()

Output publik <U> diurutkanUniqueIndices ()