SparseSegmentMeanGradV2

classe finale pubblica SparseSegmentMeanGradV2

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>
create ( ambito ambito , operando <T> grad, operando <U> indici, operando <V> segmentId, operando <Integer> denseOutputDim0)
Metodo factory per creare una classe che racchiude una nuova operazione SparseSegmentMeanGradV2.
Uscita <T>
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

Uscita pubblica <T> uscita ()

output pubblico <U> sortedUniqueIndices ()