テンソルフロー::作戦::トップK

#include <nn_ops.h>

最後の次元の最大k個の要素の値とインデックスを検索します。

まとめ

入力がベクトル (ランク 1) の場合、ベクトル内で最大のk個のエントリを検索し、それらの値とインデックスをベクトルとして出力します。したがって、 values[j] input内のj番目に大きいエントリであり、そのインデックスはindices[j]です。

行列 (それぞれ上位の入力) の場合、各行の上位k個のエントリ (それぞれ最後の次元に沿ったベクトル) を計算します。したがって、

values.shape = indices.shape = input.shape[:-1] + [k]

2 つの要素が等しい場合、インデックスの小さい要素が最初に表示されます。

引数:

  • スコープ:スコープオブジェクト
  • 入力: 最後の次元が少なくともk 1 次元以上。
  • k: 0-D。最後の次元に沿って (行列の各行に沿って) 検索する上位要素の数。

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

  • sorted: true の場合、結果として得られるk個の要素が値によって降順に並べ替えられます。

戻り値:

  • Output値: 最後の次元の各スライスに沿った最大k個の要素。
  • Outputインデックス: inputの最後の次元内のvaluesのインデックス。

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

TopK (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input k)
TopK (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input k, const TopK::Attrs & attrs)

パブリック属性

indices
operation
values

パブリック静的関数

Sorted (bool x)

構造体

tensorflow:: ops:: TopK:: Attrs

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

パブリック属性

インデックス

::tensorflow::Output indices

手術

Operation operation

価値観

::tensorflow::Output values

公共機能

トップK

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k
)

トップK

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k,
  const TopK::Attrs & attrs
)

パブリック静的関数

ソート済み

Attrs Sorted(
  bool x
)