Restituisce i valori k min/max e i relativi indici dell'operando di input in modo approssimativo.
Vedi https://arxiv.org/abs/2206.14286 per i dettagli dell'algoritmo. Questa operazione è attualmente ottimizzata solo su TPU.
Classi nidificate
classe | CircaTopK.Options | Attributi facoltativi per ApproxTopK |
Metodi pubblici
approxTopK.Options statico | aggregateToTopk (aggregatoBooleanoToTopk) |
statico <T estende il numero> approxTopK <T> | crea ( ambito ambito , operando <T> input, Long k, Opzioni... opzioni) Metodo factory per creare una classe che racchiude una nuova operazione approxTopK. |
Output <Intero> | indici () Gli indici dei "valori" lungo la "dimensione_riduzione" dell'operando "input". |
approxTopK.Options statico | isMaxK (valore booleano isMaxK) |
approxTopK.Options statico | richiamoTarget (richiamo target mobile) |
approxTopK.Options statico | riduzioneDimensione (Riduzionedimensione lunga) |
approxTopK.Options statico | riduzioneInputSizeOverride (riduzione lungaInputSizeOverride) |
Uscita <T> | valori () I valori k min/max lungo "reduction_dimension" dell'operando "input". |
Metodi ereditati
Metodi pubblici
public static ApprossimaTopK.Options aggregateToTopk (Boolean aggregateToTopk)
Parametri
aggregatoToTopk | Se vero, aggrega i risultati approssimativi a top-k. Se falso, restituisce i risultati approssimativi. Il numero dei risultati approssimativi è definito dall'implementazione ed è maggiore del valore "k" specificato. |
---|
public static ApprossimaTopK <T> create ( ambito ambito , operando <T> input, Long k, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione approxTopK.
Parametri
scopo | ambito attuale |
---|---|
ingresso | Matrice da cercare. Deve essere almeno 1-D di tipo mobile |
K | Specifica il numero di min/max-k. |
opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di approxTopK
public Output <Integer> indici ()
Gli indici dei "valori" lungo la "dimensione_riduzione" dell'operando "input".
public static ApprossimaTopK.Options isMaxK (Booleano isMaxK)
Parametri
èMaxK | Se vero, calcola max-k; altrimenti calcola min-k. |
---|
public static ApprossimaTopK.OptionscallTarget ( FloatcallTarget)
Parametri
richiamoTarget | Richiamare l'obiettivo per l'approssimazione. Intervallo in (0,1] |
---|
public static ApprossimaTopK.Options riduzioneDimensione (Lunga riduzioneDimensione)
Parametri
riduzioneDimensione | Dimensione intera lungo la quale effettuare la ricerca. Impostazione predefinita: -1. |
---|
public static ApprossimaTopK.Options riduzioneInputSizeOverride (riduzione lungaInputSizeOverride)
Parametri
riduzioneInputSizeOverride | Se impostato su un valore positivo, sovrascrive la dimensione determinata da "input[reduction_dim]" per valutare il richiamo. Questa opzione è utile quando l'"input" fornito è solo un sottoinsieme del calcolo complessivo in SPMD o pipeline distribuite, dove la dimensione effettiva dell'input non può essere differita dalla forma dell'"input". |
---|
valori di output <T> pubblici ()
I valori k min/max lungo "reduction_dimension" dell'operando "input". Le dimensioni sono le stesse dell'operando "input" tranne che per "reduction_dimension": quando "aggregate_to_topk" è vero, la dimensione di riduzione è "k"; in caso contrario, è maggiore uguale a "k" dove la dimensione è definita dall'implementazione.