SparseSegmentSumGradV2

classe finale publique SparseSegmentSumGradV2

Calcule les dégradés pour SparseSegmentSum.

Renvoie la "sortie" du tenseur avec la même forme que grad, à l'exception de la dimension 0 dont la valeur est le nombre d'index uniques dans "indices". Renvoie également le vecteur "sorted_unique_indices" contenant les index correspondants de "indices".

Méthodes publiques

static <T étend le nombre, U étend le nombre, V étend le nombre> SparseSegmentSumGradV2 <T, U>
créer ( Portée de la portée , Opérande <T> grad, Opérande <U> indices, Opérande <V> segmentIds, Opérande <Integer> denseOutputDim0)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération SparseSegmentSumGradV2.
Sortie <T>
sortir ()
Sortie <U>

Méthodes héritées

Méthodes publiques

public static SparseSegmentSumGradV2 <T, U> créer ( Portée de portée , Opérande <T> grad, Opérande <U> indices, Opérande <V> segmentIds, Opérande <Integer> denseOutputDim0)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération SparseSegmentSumGradV2.

Paramètres
portée portée actuelle
diplômé gradient propagé à l'opération SparseSegmentSum.
indices indices transmis à l’opération SparseSegmentSum correspondante.
ID de segment segment_ids transmis à l’opération SparseSegmentSum correspondante.
denseOutputDim0 dimension 0 des "données" transmises à SparseSegmentSum op.
Retour
  • une nouvelle instance de SparseSegmentSumGradV2

sortie publique <T> sortie ()

sortie publique <U> triéeUniqueIndices ()