このページは Cloud Translation API によって翻訳されました。
Switch to English

tensorflow :: ops :: TopK

#include <nn_ops.h>

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

概要

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

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

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

2つの要素が等しい場合、lower-index要素が最初に表示されます。

引数:

  • scope: Scopeオブジェクト
  • 入力:1次元以上、最後の次元は少なくともk
  • 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

公開機能

TopK

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

TopK

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

パブリック静的関数

並べ替え

Attrs Sorted(
  bool x
)