flux tensoriel : : opérations : : Somme du segment non trié
#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 des tuples j...
tel que segment_ids[j...] == i
. Contrairement à SegmentSum
, segment_ids
n'a pas besoin d'être trié et ne doit pas nécessairement couvrir toutes les valeurs dans la plage 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é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 dimensionssegment_ids.rank
, qui sont remplacées par une seule dimension de taillenum_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
sortir
::tensorflow::Output output
Fonctions publiques
Somme du segment non trié
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