tensorflow:: ops:: TopK

最後の次元におけるk個の最大要素の値とインデックスを求めます。

まとめ

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

行列（または高ランク入力）の場合、各行（または最後の次元に沿ったベクトル）の上位k個のエントリを計算します。したがって、

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

2つの要素が同じ場合、インデックスの小さい方の要素が先に表示されます。

議論:

• スコープ:スコープオブジェクト
• 入力: 1 次元以上で、最後の次元は少なくともk 。
• 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
)