パブリック静的クラスUpperTopK.Options
ApproxTopK
のオプションの属性
パブリックメソッド
おおよそのTopK.オプション | aggregateToTopk (ブール値のaggregateToTopk) |
おおよそのTopK.オプション | isMaxK (ブール値 isMaxK) |
おおよそのTopK.オプション | rememberTarget (浮動小数点リコールターゲット) |
おおよそのTopK.オプション | 縮小寸法(長い縮小寸法) |
おおよそのTopK.オプション | ductionInputSizeOverride (長いreductionInputSizeOverride) |
継承されたメソッド
パブリックメソッド
public概説TopK.Options aggregateToTopk (ブール値aggregateToTopk)
パラメーター
集計からTopkまで | true の場合、結果を上位 k に近似して集計します。 false の場合、近似結果を返します。近似結果の数は実装で定義され、指定された `k` より大きくなります。 |
---|
public UpperK.Options isMaxK (ブール値 isMaxK)
パラメーター
isMaxK | true の場合、max-k を計算します。それ以外の場合は min-k を計算します。 |
---|
public AverageTopK.Options rememberTarget (浮動小数点リコールターゲット)
パラメーター
リコールターゲット | 近似のターゲットを呼び出します。 (0,1] の範囲 |
---|
public UpperK.OptionsductionInputSizeOverride ( LongductionInputSizeOverride)
パラメーター
ductionInputSizeOverride | 正の値に設定すると、リコールを評価するために `input[reduction_dim]` によって決定されたサイズがオーバーライドされます。このオプションは、指定された `input` が SPMD または分散パイプラインの計算全体のサブセットにすぎず、実際の入力サイズが `input` 形状によって延期できない場合に便利です。 |
---|