tensorflow :: opérations :: UnsortedSegmentSum

#include <math_ops.h>

Calcule la somme le long des segments d'un tenseur.

Résumé

Lisez la section sur la segmentation pour une explication des segments.

Calcule un tenseur tel que \(output[i] = {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 identifiant de segment i , la 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]]

Arguments:

  • scope: un objet Scope
  • segment_ids: un tenseur dont la forme est un préfixe de data.shape .

Retour:

  • Output : 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 .

Constructeurs et destructeurs

UnsortedSegmentSum (const :: tensorflow::Scope & scope, :: tensorflow::Input data, :: tensorflow::Input segment_ids, :: tensorflow::Input num_segments)

Attributs publics

operation
output

Fonctions publiques

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Attributs publics

opération

Operation operation

production

::tensorflow::Output output

Fonctions publiques

UnsortedSegmentSum

 UnsortedSegmentSum(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input data,
  ::tensorflow::Input segment_ids,
  ::tensorflow::Input num_segments
)

nœud

::tensorflow::Node * node() const 

operator :: tensorflow :: Input

 operator::tensorflow::Input() const 

operator :: tensorflow :: Output

 operator::tensorflow::Output() const