パブリックファイナルクラス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> | |
| 出力< 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 の新しいインスタンス