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> | |
Keluaran <T> | keluaran () |
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