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

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.

Retours :

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

Attributs publics

Operation operation

::tensorflow::Output values

 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
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs Reverse(
  bool x
)