tensorflow :: opérations :: TopK

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

Résumé

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

Pour les matrices (respectivement 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 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 facultatifs (voir Attrs ):

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

Retour:

• Valeurs de Output : Les k plus grands éléments le long de chaque dernière tranche dimensionnelle.
• Indices de Output : Les indices de values dans la dernière dimension d' 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
)