SparseSegmentSqrtNGradV2

classe finale pubblica SparseSegmentSqrtNGradV2

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>
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 SparseSegmentSqrtNGradV2.
Uscita <T>
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

Uscita pubblica <T> uscita ()

output pubblico <U> sortedUniqueIndices ()