Calcola i gradienti per SparseSegmentMean.
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> SparseSegmentMeanGradV2 <T, U> | |
Uscita <T> | produzione () |
Uscita <U> |
Metodi ereditati
Metodi pubblici
public static SparseSegmentMeanGradV2 <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 SparseSegmentMeanGradV2.
Parametri
scopo | ambito attuale |
---|---|
grado | gradiente propagato allo SparseSegmentMean op. |
indici | indici passati al corrispondente SparseSegmentMean op. |
segmentId | segment_ids passato al corrispondente SparseSegmentMean op. |
densoOutputDim0 | dimensione 0 dei "dati" passati a SparseSegmentMean op. |
ritorna
- una nuova istanza di SparseSegmentMeanGradV2