tensorflow :: opérations :: TopK

#include <nn_ops.h>

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 .

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)

Structs

tensorflow :: ops :: TopK :: Attrs

Setters d'attributs en option 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
)