Calcola i gradienti per SparseSegmentSqrtN.
Restituisce il tensore "output" con la stessa forma di grad, ad eccezione della dimensione 0 il cui valore è il numero di indici univoci in "indices". Restituisce anche il vettore "sorted_unique_indices" contenente gli indici corrispondenti da "indices".
Metodi pubblici
statico <T estende il numero, U estende il numero, V estende il numero> SparseSegmentSqrtNGradV2 <T, U> | |
Uscita <T> | produzione () |
Uscita <U> |
Metodi ereditati
Metodi pubblici
public static SparseSegmentSqrtNGradV2 <T, U> create ( scope scope, operando <T> grad, operando <U> indici, operando <V> segmentIds, operando <Integer> denseOutputDim0)
Metodo factory per creare una classe che racchiude una nuova operazione SparseSegmentSqrtNGradV2.
Parametri
scopo | ambito attuale |
---|---|
grado | gradiente propagato allo SparseSegmentSqrtN op. |
indici | indici passati al corrispondente SparseSegmentSqrtN op. |
segmentId | segment_ids passato al corrispondente SparseSegmentSqrtN op. |
densoOutputDim0 | dimensione 0 dei "dati" passati a SparseSegmentSqrtN op. |
ritorna
- una nuova istanza di SparseSegmentSqrtNGradV2