テンソルフロー::作戦:: NthElement

最後の次元のn次統計量の値を検索します。

まとめ

入力がベクトル (ランク 1) の場合、ベクトル内で n 番目に小さい値であるエントリを検索し、その値をスカラー テンソルとして出力します。

行列 (それぞれ、より高いランクの入力) の場合、各行の n 番目に小さい値であるエントリ (それぞれ、最後の次元に沿ったベクトル) を計算します。したがって、

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

引数:

• スコープ:スコープオブジェクト
• 入力: 1 次元以上、最後の次元が少なくともn+1 。
• n: 0-D。最後の次元に沿って (行列の各行に沿って) 選択する並べ替えられたベクトルの位置。 n の有効範囲は[0, input.shape[:-1])

オプションの属性 ( Attrsを参照):

• reverse: True に設定すると、ベクトル内の n 番目に大きい値が検索され、その逆も同様です。

戻り値:

• Output : 最後の次元の各スライスに沿ったn次の統計量。

パブリック属性

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
)