flux tensoriel : : opérations : : Non triéSegmentMax

Calcule le maximum le long des segments d'un tenseur.

Résumé

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

Cet opérateur est similaire à l'opérateur de somme de segments non triés trouvé (ici) . Au lieu de calculer la somme sur les segments, il calcule le maximum tel que :

\(output_i = {j...} data[j...]\) où max est supérieur aux tuples j... tels que segment_ids[j...] == i .

Si le maximum est vide pour un ID de segment donné i , il génère la plus petite valeur possible pour le type numérique spécifique, output[i] = numeric_limits ::lowest() output[i] = numeric_limits ::lowest() .

Si l'ID de segment donné i est négatif, alors la valeur correspondante est supprimée et ne sera pas incluse dans le résultat.

Par exemple:

c = tf.constant([[1,2,3,4], [5,6,7,8], [4,3,2,1]])
tf.unsorted_segment_max(c, tf.constant([0, 1, 0]), num_segments=2)
# ==> [[ 4,  3, 3, 4],
#       [5,  6, 7, 8]]

Arguments :

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

Retours :

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

Attributs publics

Operation operation

::tensorflow::Output output

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

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const