パブリック最終クラス
入力オペランドの最小/最大 k 値とそのインデックスを近似的に返します。
アルゴリズムの詳細については、https://arxiv.org/abs/2206.14286 を参照してください。この操作は現在 TPU でのみ最適化されています。
ネストされたクラス
| クラス | おおよそのTopK.オプション | ApproxTopKのオプションの属性 | |
パブリックメソッド
| 静的AverageTopK.Options | aggregateToTopk (ブール値のaggregateToTopk) |
| static <T extends Number>おおよそのTopK <T> | |
| 出力<整数> | インデックス() 「input」オペランドの「reduction_dimension」に沿った「values」のインデックス。 |
| 静的AverageTopK.Options | isMaxK (ブール値 isMaxK) |
| 静的AverageTopK.Options | rememberTarget (浮動小数点リコールターゲット) |
| 静的AverageTopK.Options | 縮小寸法(長い縮小寸法) |
| 静的AverageTopK.Options | ductionInputSizeOverride (長いreductionInputSizeOverride) |
| 出力<T> | 値() 「input」オペランドの「reduction_dimension」に沿った最小/最大 k 値。 |
継承されたメソッド
パブリックメソッド
public static UpperTopK.Options aggregateToTopk (ブール値のaggregateToTopk)
パラメーター
| 集計からTopkまで | true の場合、結果を上位 k に近似して集計します。 false の場合、近似結果を返します。近似結果の数は実装で定義され、指定された `k` より大きくなります。 |
|---|
public static AverageTopK <T> create (スコープスコープ、オペランド<T> 入力、Long k、オプション...オプション)
新しい AverageTopK 操作をラップするクラスを作成するファクトリ メソッド。
パラメーター
| 範囲 | 現在のスコープ |
|---|---|
| 入力 | 検索する配列。浮動型の 1-D 以上である必要があります |
| k | min/max-k の数を指定します。 |
| オプション | オプションの属性値を持ちます |
戻り値
- UpperTopK の新しいインスタンス
public static UpperTopK.Options isMaxK (ブール値 isMaxK)
パラメーター
| isMaxK | true の場合、max-k を計算します。それ以外の場合は min-k を計算します。 |
|---|
public static AverageTopK.Options rememberTarget (浮動小数点リコールターゲット)
パラメーター
| リコールターゲット | 近似のターゲットを呼び出します。 (0,1] の範囲 |
|---|
public static AverageTopK.OptionsductionDimension ( LongductionDimension)
パラメーター
| 削減寸法 | 検索する整数の次元。デフォルト: -1。 |
|---|
public static UpperTopK.OptionsductionInputSizeOverride ( LongductionInputSizeOverride)
パラメーター
| ductionInputSizeOverride | 正の値に設定すると、リコールを評価するために `input[reduction_dim]` によって決定されたサイズがオーバーライドされます。このオプションは、指定された `input` が SPMD または分散パイプラインの計算全体のサブセットにすぎず、実際の入力サイズが `input` 形状によって延期できない場合に役立ちます。 |
|---|
public出力<T>値()
「input」オペランドの「reduction_dimension」に沿った最小/最大 k 値。次元は、`reduction_dimension` を除き、`input` オペランドと同じです。 `aggregate_to_topk` が true の場合、リダクション次元は `k` です。それ以外の場合、サイズは実装で定義される「k」より大きくなります。