Calcule la somme le long de segments clairsemés d'un tenseur.
Lisez [la section sur la segmentation](https://tensorflow.org/api_docs/python/tf/math#Segmentation) pour une explication des segments.
Comme `SegmentSum`, mais `segment_ids` peut avoir un rang inférieur à la première dimension de `data`, en sélectionnant un sous-ensemble de dimension 0, spécifié par `indices`.
Par exemple:
c = tf.constant([[1,2,3,4], [-1,-2,-3,-4], [5,6,7,8]])
# Select two rows, one segment.
tf.sparse_segment_sum(c, tf.constant([0, 1]), tf.constant([0, 0]))
# => [[0 0 0 0]]
# Select two rows, two segment.
tf.sparse_segment_sum(c, tf.constant([0, 1]), tf.constant([0, 1]))
# => [[ 1 2 3 4]
# [-1 -2 -3 -4]]
# Select all rows, two segments.
tf.sparse_segment_sum(c, tf.constant([0, 1, 2]), tf.constant([0, 0, 1]))
# => [[0 0 0 0]
# [5 6 7 8]]
# Which is equivalent to:
tf.segment_sum(c, tf.constant([0, 0, 1]))
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 TNumber > SparseSegmentSum <T> | |
Sortie <T> | sortir () A la même forme que les données, sauf pour la dimension 0 qui a la taille « k », le nombre de 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 SparseSegmentSum <T> créer ( portée de la portée, données de l'opérande <T>, opérande <? étend TNumber > indices, opérande <? étend TNumber > segmentIds)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération SparseSegmentSum.
Paramètres
portée | portée actuelle |
---|---|
indices | Un tenseur 1-D. A le même rang que `segment_ids`. |
ID de segment | Un tenseur 1-D. Les valeurs doivent être triées et peuvent être répétées. |
Retours
- une nouvelle instance de SparseSegmentSum
sortie publique <T> sortie ()
A la même forme que les données, sauf pour la dimension 0 qui a la taille « k », le nombre de segments.