tensorflow :: opérations :: Non triéSegmentProd

Calcule le produit 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 segment non trié trouvé (ici) . Au lieu de calculer la somme sur les segments, il calcule le produit de toutes les entrées appartenant à un segment tel que:

\(output_i = {j...} data[j...]\) où le produit est sur les tuples j... tels que segment_ids[j...] == i .

Par example:

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

S'il n'y a pas d'entrée pour un ID de segment donné i , il génère 1.

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.

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 .

Attributs publics

Operation operation

::tensorflow::Output output

 UnsortedSegmentProd(
  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