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

最後の次元の最大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のインデックス。

パブリック属性

::tensorflow::Output indices

Operation operation

::tensorflow::Output 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
)

Attrs Sorted(
  bool x
)