flux tensoriel : : opérations : : TopK

#include <nn_ops.h>

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 .

Constructeurs et Destructeurs

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)

Attributs publics

indices
operation
values

Fonctions statiques publiques

Sorted (bool x)

Structures

tensorflow :: ops :: TopK :: Attrs

Setters d'attributs facultatifs pour TopK .

Attributs publics

indices 

::tensorflow::Output indices

opération 

Operation operation

valeurs 

::tensorflow::Output values

Fonctions publiques

TopK 

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

TopK 

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

Fonctions statiques publiques

Trié 

Attrs Sorted(
  bool x
)