publiczne zajęcia końcowe ApproxTopK
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.