Cette page a été traduite par l'API Cloud Translation.
Switch to English

tensorflow :: ops :: UnsortedSegmentSum

#include <math_ops.h>

Calcule la somme des segments le long 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 ...] \) lorsque la somme est supérieure à tuples j... de telle sorte que segment_ids[j...] == i . Contrairement à SegmentSum , segment_ids ne doivent pas être classés et ne doivent pas couvrir toutes les valeurs de la gamme complète des valeurs valides.

Si la somme est vide pour un ID de segment donné i , output[i] = 0 . Si l'ID de segment donné i est négatif, la valeur est supprimée et ne sera pas ajouté à la somme du segment.

num_segments doit être égal au nombre d'ID de segment distinct.

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:

  • portée: A Portée objet
  • segment_ids: tenseur de A dont la forme est un préfixe de data.shape .

Retour:

  • Output : A une même forme que des données, sauf pour les premiers segment_ids.rank dimensions, qui sont remplacées par une seule dimension , qui a une 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

Les fonctions publiques

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

attributs publics

opération

 Operation operation

production

:: tensorflow::Output output

Les 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 

opérateur :: tensorflow :: entrée

 operator::tensorflow::Input() const 

opérateur :: tensorflow :: sortie

 operator::tensorflow::Output() const