Renvoie les valeurs k min/max et leurs indices de l'opérande d'entrée de manière approximative.
Voir https://arxiv.org/abs/2206.14286 pour les détails de l'algorithme. Cette opération n'est actuellement optimisée que sur TPU.
Classes imbriquées
classe | EnvironTopK.Options | Attributs facultatifs pour ApproxTopK |
Méthodes publiques
statique ApproxTopK.Options | globalVersTopk (agrégat booléenVersTopk) |
statique <T étend le nombre> ApproxTopK <T> | créer ( Portée de portée , Entrée Opérande <T>, Long k, Options... options) Méthode d'usine pour créer une classe encapsulant une nouvelle opération ApproxTopK. |
Sortie <Entier> | indices () Les indices des « valeurs » le long de la « dimension_de réduction » de l'opérande « d'entrée ». |
statique ApproxTopK.Options | isMaxK (isMaxK booléen) |
statique ApproxTopK.Options | rappelTarget (rappel flottantTarget) |
statique ApproxTopK.Options | réductionDimension (Longue réductionDimension) |
statique ApproxTopK.Options | réductionInputSizeOverride (longue réductionInputSizeOverride) |
Sortie <T> | valeurs () Les valeurs k min/max le long de la « dimension_de réduction » de l'opérande « d'entrée ». |
Méthodes héritées
Méthodes publiques
public statique 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 static ApproxTopK <T> créer (portée de portée , entrée d'opérande <T>, k long, options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération ApproxTopK.
Paramètres
portée | portée actuelle |
---|---|
saisir | Tableau à rechercher. Doit être au moins 1-D de type flottant |
k | Spécifie le nombre de min/max-k. |
choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance d'ApproxTopK
public Output <Integer> indices ()
Les indices des « valeurs » le long de la « dimension_de réduction » de l'opérande « d'entrée ».
public statique ApproxTopK.Options isMaxK (booléen isMaxK)
Paramètres
estMaxK | Lorsque c'est vrai, calcule max-k ; sinon, calcule min-k. |
---|
public statique ApproxTopK.Options rappelTarget (Float rappelTarget)
Paramètres
rappelCible | Rappel de la cible pour l'approximation. Plage en (0,1] |
---|
public statique 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 statique 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 ». |
---|
valeurs de sortie publiques <T> ()
Les valeurs k min/max le long de la « dimension_de réduction » de l'opérande « d'entrée ». Les dimensions sont les mêmes que l'opérande `input` sauf pour la `reduction_dimension` : lorsque `aggregate_to_topk` est vrai, la dimension de réduction est `k` ; sinon, sa valeur supérieure est égale à « k » où la taille est définie par l'implémentation.