Zwraca wartości min/max k i ich indeksy argumentu wejściowego w przybliżony sposób.
Szczegóły algorytmu można znaleźć na stronie https://arxiv.org/abs/2206.14286. Ta opcja jest obecnie zoptymalizowana tylko pod kątem TPU.
Klasy zagnieżdżone
klasa | OkołoTopK.Opcji | Opcjonalne atrybuty dla ApproxTopK |
Metody publiczne
statyczne OkołoTopK.Opcje | agregatToTopk (logiczny agregatToTopk) |
statyczny <T liczba rozszerzeń> OkołoTopK <T> | |
Wyjście <liczba całkowita> | indeksy () Indeksy „wartości” wzdłuż „wymiaru_redukcji” argumentu „wejściowego”. |
statyczne OkołoTopK.Opcje | isMaxK (Boolean isMaxK) |
statyczne OkołoTopK.Opcje | przywołanie celu (pływające przywołanie celu) |
statyczne OkołoTopK.Opcje | wymiar redukcji (wymiar redukcji długiej) |
statyczne OkołoTopK.Opcje | redukcjaInputSizeOverride (długa redukcjaInputSizeOverride) |
Wyjście <T> | wartości () Wartości min/max k wzdłuż „wymiaru_redukcji” argumentu „input”. |
Metody dziedziczone
Metody publiczne
publiczny statyczny ApproxTopK.Options mergeToTopk (boolowski agregat ToTopk)
Parametry
agregatDoTopk | Jeśli ma wartość true, agreguje przybliżone wyniki do góry-k. Jeśli false, zwraca przybliżone wyniki. Liczba przybliżonych wyników jest zdefiniowana w implementacji i jest większa niż określone „k”. |
---|
public static ApproxTopK <T> create (zakres zakresu , wejście argumentu <T>, długie k, opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację ApproxTopK.
Parametry
zakres | aktualny zakres |
---|---|
wejście | Tablica do przeszukania. Musi mieć co najmniej 1-D typu zmiennoprzecinkowego |
k | Określa liczbę min/max-k. |
opcje | przenosi opcjonalne wartości atrybutów |
Zwroty
- nowa instancja ApproxTopK
public Output <Integer> indeksy ()
Indeksy „wartości” wzdłuż „wymiaru_redukcji” argumentu „wejściowego”.
public static ApproxTopK.Options isMaxK (Boolean isMaxK)
Parametry
jestMaxK | Jeśli ma wartość true, oblicza max-k; w przeciwnym razie oblicza min-k. |
---|
public static ApproxTopK.Options RecallTarget (Float RecallTarget)
Parametry
Przypomnij sobie cel | Przywołaj cel do przybliżenia. Zakres w (0,1] |
---|
public static ApproxTopK.Options redukcjaWymiar (Długa redukcjaWymiar)
Parametry
redukcjaWymiar | Wymiar całkowity, wzdłuż którego należy szukać. Wartość domyślna: -1. |
---|
public static ApproxTopK.Options redukcjaInputSizeOverride (długa redukcjaInputSizeOverride)
Parametry
redukcjaInputSizeOverride | Gdy jest ustawiony na wartość dodatnią, zastępuje rozmiar określony przez `input[reduction_dim]` na potrzeby oceny wycofania. Ta opcja jest użyteczna, gdy dane dane wejściowe stanowią jedynie podzbiór ogólnych obliczeń w SPMD lub potokach rozproszonych, gdzie prawdziwego rozmiaru danych wejściowych nie można odroczyć za pomocą kształtu danych wejściowych. |
---|
publiczne wartości wyjściowe <T> ()
Wartości min/max k wzdłuż „wymiaru_redukcji” argumentu „input”. Wymiar jest taki sam jak operand `input`, z wyjątkiem `wymiaru_redukcji`: gdy `agregat_do_topk` ma wartość true, wymiarem redukcji jest `k`; w przeciwnym razie większa jest równa `k`, gdzie rozmiar jest zdefiniowany w implementacji.