TopK

パブリックファイナルクラスTopK

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

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

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

値.形状 = インデックス.形状 = 入力.形状[:-1] + [k]

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

ネストされたクラス

クラスTopK.オプションTopKのオプションの属性

定数

弦OP_NAME TensorFlow コア エンジンによって認識される、この演算の名前

パブリックメソッド

static <T extends TNumber > TopK <T>
create (スコープscope、オペランド<T>入力、オペランド<TInt32> k、オプション...オプション)
新しい TopK オペレーションをラップするクラスを作成するファクトリ メソッド。
出力< TInt32 >
インデックス()
「input」の最後の次元内の「values」のインデックス。
静的TopK.Options
ソート済み(ブールソート済み)
出力<T>
値()
各最後の次元のスライスに沿った最大の「k」個の要素。

継承されたメソッド

定数

パブリック静的最終文字列OP_NAME

TensorFlow コア エンジンによって認識される、この演算の名前

定数値: 「TopKV2」

パブリックメソッド

public static TopK <T> create (スコープscope、オペランド<T>入力、オペランド<TInt32> k、オプション...オプション)

新しい TopK オペレーションをラップするクラスを作成するファクトリ メソッド。

パラメーター
範囲現在のスコープ
入力1 次元以上、最後の次元が少なくとも `k` であること。
k 0-D。最後の次元に沿って (行列の各行に沿って) 検索する上位要素の数。
オプションオプションの属性値を持ちます
戻り値
  • TopK の新しいインスタンス

public出力< TInt32 >インデックス()

「input」の最後の次元内の「values」のインデックス。

public static TopK.Optionsソート(ブール値ソート)

パラメーター
並べ替えられたtrue の場合、結果として得られる `k` 要素は値によって降順に並べ替えられます。

public出力<T>値()

各最後の次元のスライスに沿った最大の「k」個の要素。