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

#include <nn_ops.h>

最後の次元の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次の統計量。

コンストラクターとデストラクター

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)

パブリック属性

operation
values

公共機能

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

パブリック静的関数

Reverse (bool x)

構造体

tensorflow:: ops:: NthElement:: Attrs

NthElementのオプションの属性セッター。

パブリック属性

手術

Operation operation

価値観

::tensorflow::Output values

公共機能

NthElement

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

NthElement

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

ノード

::tensorflow::Node * node() const 

演算子::tensorflow::入力

 operator::tensorflow::Input() const 

演算子::tensorflow::出力

 operator::tensorflow::Output() const 

パブリック静的関数

逆行する

Attrs Reverse(
  bool x
)