tensorflow::ops::TopK

Trouve les valeurs et les indices des k plus grands éléments pour la dernière dimension.

Résumé

Si l'entrée est un vecteur (de rang 1), la fonction trouve les k plus grandes valeurs du vecteur et renvoie leurs valeurs et indices sous forme de vecteurs. Ainsi, values[j] correspond à la j -ième plus grande valeur de input , et son indice est indices[j] .

Pour les matrices (resp. les entrées de rang supérieur), calcule les k premiers éléments de chaque ligne (resp. du vecteur le long de la dernière dimension). Ainsi,

values.shape = indices.shape = input.shape[:-1] + [k]

Si deux éléments sont égaux, l'élément ayant l'indice le plus bas apparaît en premier.

Arguments :

• portée : un objet Scope
• entrée : 1D ou plus avec la dernière dimension au moins k .
• k : 0-D. Nombre d'éléments supérieurs à rechercher le long de la dernière dimension (le long de chaque ligne pour les matrices).

Attributs optionnels (voir Attrs ) :

• sorted : Si la valeur est vraie, les k éléments résultants seront triés par valeurs dans l'ordre décroissant.

Retours :

• Valeurs Output : Les k plus grands éléments le long de chaque dernière tranche dimensionnelle.
• Indices Output : Les indices des values dans la dernière dimension de input .

attributs publics

::tensorflow::Output indices

Operation operation

::tensorflow::Output values

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k
)

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k,
  const TopK::Attrs & attrs
)

Attrs Sorted(
  bool x
)