ApproxTopK.Options

classe statique publique ApproxTopK.Options

Attributs facultatifs pour ApproxTopK

Méthodes publiques

EnvironTopK.Options
globalVersTopk (agrégat booléenVersTopk)
EnvironTopK.Options
isMaxK (isMaxK booléen)
EnvironTopK.Options
rappelTarget (rappel flottantTarget)
EnvironTopK.Options
réductionDimension (Longue réductionDimension)
EnvironTopK.Options
réductionInputSizeOverride (longue réductionInputSizeOverride)

Méthodes héritées

Méthodes publiques

public ApproxTopK.Options AggregateToTopk (agrégat booléenToTopk)

Paramètres
agrégatVersTopk Lorsque cela est vrai, les résultats se rapprochent du top-k. Lorsque faux, renvoie les résultats approximatifs. Le nombre de résultats approximatifs est défini par l'implémentation et est supérieur au « k » spécifié.

public ApproxTopK.Options isMaxK (booléen isMaxK)

Paramètres
estMaxK Lorsque c'est vrai, calcule max-k ; sinon, calcule min-k.

public ApproxTopK.Options rappelTarget (Float rappelTarget)

Paramètres
rappelCible Rappel de la cible pour l'approximation. Plage en (0,1]

public ApproxTopK.Options réductionDimension (Long réductionDimension)

Paramètres
réductionDimension Dimension entière le long de laquelle effectuer la recherche. Par défaut : -1.

public ApproxTopK.Options réductionInputSizeOverride (longue réductionInputSizeOverride)

Paramètres
réductionInputSizeOverride Lorsqu'il est défini sur une valeur positive, il remplace la taille déterminée par `input[reduction_dim]` pour évaluer le rappel. Cette option est utile lorsque l'« entrée » donnée n'est qu'un sous-ensemble du calcul global dans SPMD ou les pipelines distribués, où la taille réelle de l'entrée ne peut pas être différée par la forme de l'« entrée ».