Menghitung gradien untuk SparseSegmentSqrtN.
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> SparseSegmentSqrtNGradV2 <T, U> | |
Keluaran <T> | keluaran () |
Keluaran <U> |
Metode Warisan
Metode Publik
public static SparseSegmentSqrtNGradV2 <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 SparseSegmentSqrtNGradV2 baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
lulusan | gradien disebarkan ke operasi SparseSegmentSqrtN. |
indeks | indeks diteruskan ke operasi SparseSegmentSqrtN yang sesuai. |
segmenId | segment_ids diteruskan ke operasi SparseSegmentSqrtN yang sesuai. |
padatOutputDim0 | dimensi 0 dari "data" diteruskan ke operasi SparseSegmentSqrtN. |
Kembali
- contoh baru dari SparseSegmentSqrtNGradV2