flux tensoriel : : opérations : : NièmeÉlément

#include <nn_ops.h>

Recherche les valeurs de la statistique d'ordre n pour la dernière dimension.

Résumé

Si l'entrée est un vecteur (rang 1), recherche les entrées qui sont la nième plus petite valeur du vecteur et génère leurs valeurs sous forme de tenseur scalaire.

Pour les matrices (resp. entrée de rang supérieur), calcule les entrées qui sont la nième plus petite valeur de chaque ligne (resp. vecteur le long de la dernière dimension). Ainsi,

values.shape = input.shape[:-1]

Arguments:

  • scope : un objet Scope
  • entrée : 1-D ou supérieur avec la dernière dimension au moins n+1 .
  • n : 0-D. Position du vecteur trié à sélectionner le long de la dernière dimension (le long de chaque ligne pour les matrices). La plage valide de n est [0, input.shape[:-1])

Attributs facultatifs (voir Attrs ) :

  • reverse : lorsque la valeur est True, recherchez la nième plus grande valeur du vecteur et vice versa.

Retour:

  • Output : la statistique d'ordre n le long de chaque dernière tranche dimensionnelle.

Constructeurs et Destructeurs

NthElement (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input n)
NthElement (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input n, const NthElement::Attrs & attrs)

Attributs publics

operation
values

Fonctions publiques

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Fonctions statiques publiques

Reverse (bool x)

Structures

tensorflow :: ops :: NthElement :: Attrs

Setters d'attributs facultatifs pour NthElement .

Attributs publics

opération

Operation operation

valeurs

::tensorflow::Output values

Fonctions publiques

NièmeÉlément

 NthElement(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input n
)

NièmeÉlément

 NthElement(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input n,
  const NthElement::Attrs & attrs
)

nœud

::tensorflow::Node * node() const 

opérateur :: tensorflow :: Entrée

 operator::tensorflow::Input() const 

opérateur :: tensorflow :: Sortie

 operator::tensorflow::Output() const 

Fonctions statiques publiques

Inverse

Attrs Reverse(
  bool x
)