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> | |
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