tensorflow :: opérations :: NthElement

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), trouve les entrées qui sont la nième valeur la plus petite du vecteur et génère leurs valeurs sous forme de tenseur scalaire.

Pour les matrices (respectivement entrée de rang supérieur), calcule les entrées qui sont la nième valeur la plus petite 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 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 facultatifs (voir Attrs ):

• reverse: lorsqu'il est défini sur True, recherchez la nième valeur la plus grande du vecteur et vice versa.

Retour:

• Output : la statistique d'ordre n 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
)