ApproxTopK.Options

パブリック静的クラス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.OptionsductionDimension ( LongductionDimension)

パラメーター
削減寸法検索する整数の次元。デフォルト: -1。

public UpperK.OptionsductionInputSizeOverride ( LongductionInputSizeOverride)

パラメーター
ductionInputSizeOverride正の値に設定すると、リコールを評価するために `input[reduction_dim]` によって決定されたサイズがオーバーライドされます。このオプションは、指定された `input` が SPMD または分散パイプラインの計算全体のサブセットにすぎず、実際の入力サイズが `input` 形状によって延期できない場合に便利です。