classe finale publique SparseSegmentSqrtNGradV2
Calcule les dégradés pour SparseSegmentSqrtN.
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> SparseSegmentSqrtNGradV2 <T, U> | |
| Sortie <T> | sortir () |
| Sortie <U> |
Méthodes héritées
Méthodes publiques
public static SparseSegmentSqrtNGradV2 <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 SparseSegmentSqrtNGradV2.
Paramètres
| portée | portée actuelle |
|---|---|
| diplômé | gradient propagé à l'op SparseSegmentSqrtN. |
| indices | indices transmis à l’opération SparseSegmentSqrtN correspondante. |
| ID de segment | segment_ids transmis à l’opération SparseSegmentSqrtN correspondante. |
| denseOutputDim0 | dimension 0 des "données" transmises à SparseSegmentSqrtN op. |
Retour
- une nouvelle instance de SparseSegmentSqrtNGradV2