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