flux tensoriel : : opérations : : TopK

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

Résumé

Si l'entrée est un vecteur (rang 1), recherche les k entrées les plus grandes du vecteur et génère leurs valeurs et indices sous forme de vecteurs. Ainsi, values[j] est la j -ème plus grande entrée dans input , et son index est indices[j] .

Pour les matrices (resp. entrée de rang supérieur), calcule les k premières entrées de chaque ligne (resp. 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 d’indice inférieur apparaît en premier.

Arguments :

• scope : un objet Scope
• entrée : 1-D ou supérieur 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 facultatifs (voir Attrs ) :

• trié : si vrai, les k éléments résultants seront triés par valeurs par ordre décroissant.

Retours :

• Valeurs Output : les k plus grands éléments le long de chaque dernière tranche dimensionnelle.
• Indices Output : 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
)