tensorflow::ops::NthElement

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

Résumé

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

Pour les matrices (resp. les entrées de rang supérieur), calcule les éléments qui correspondent à la n-ième plus petite valeur de chaque ligne (resp. du vecteur le long de la dernière dimension). Ainsi,

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

Arguments :

• portée : un objet Scope
• entrée : 1D ou plus 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 optionnels (voir Attrs ) :

• reverse : Si la valeur est True, cette option permet de trouver 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
)