Calcule la somme le long des segments d'un tenseur.
Lisez [la section sur la segmentation](https://tensorflow.org/api_docs/python/tf/math#Segmentation) pour une explication des segments.
Calcule un tenseur tel que \\(output[i] = \sum_{j...} data[j...]\\) où la somme est sur les tuples `j...` tels que `segment_ids[j...] == i`. Contrairement à `SegmentSum`, `segment_ids` n'a pas besoin d'être trié et n'a pas besoin de couvrir toutes les valeurs dans la plage complète des valeurs valides.
Si la somme est vide pour un ID de segment `i` donné, `output[i] = 0`. Si l'ID de segment donné « i » est négatif, la valeur est supprimée et ne sera pas ajoutée à la somme du segment.
`num_segments` doit être égal au nombre d'ID de segment distincts.
c = tf.constant([[1,2,3,4], [5,6,7,8], [4,3,2,1]])
tf.unsorted_segment_sum(c, tf.constant([0, 1, 0]), num_segments=2)
# ==> [[ 5, 5, 5, 5],
# [5, 6, 7, 8]]
Constantes
Chaîne | OP_NAME | Le nom de cette opération, tel que connu par le moteur principal TensorFlow |
Méthodes publiques
Sortie <T> | comme Sortie () Renvoie le handle symbolique du tenseur. |
statique <T étend TType > UnsortedSegmentSum <T> | |
Sortie <T> | sortir () A la même forme que les données, à l'exception des premières dimensions `segment_ids.rank`, qui sont remplacées par une seule dimension de taille `num_segments`. |
Méthodes héritées
Constantes
chaîne finale statique publique OP_NAME
Le nom de cette opération, tel que connu par le moteur principal TensorFlow
Méthodes publiques
sortie publique <T> asOutput ()
Renvoie le handle symbolique du tenseur.
Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.
public static UnsortedSegmentSum <T> créer (portée de portée , données d'opérande <T>, opérande <? extends TNumber > segmentIds, Operand <? extends TNumber > numSegments)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération UnsortedSegmentSum.
Paramètres
portée | portée actuelle |
---|---|
ID de segment | Un tenseur dont la forme est un préfixe de `data.shape`. |
Retour
- une nouvelle instance de UnsortedSegmentSum
sortie publique <T> sortie ()
A la même forme que les données, à l'exception des premières dimensions `segment_ids.rank`, qui sont remplacées par une seule dimension de taille `num_segments`.