Znajduje wartości i indeksy „k” największych elementów ostatniego wymiaru.
Jeśli danymi wejściowymi jest wektor (ranga-1), znajduje „k” największych wpisów w wektorze i wyświetla ich wartości i wskaźniki jako wektory. Zatem „wartości[j]” to „j”-ty największy wpis w „wejściowych”, a jego indeks to „indeksy[j]”.
Dla macierzy (odpowiednio danych wejściowych o wyższej randze) oblicza „k” górnych wpisów w każdym wierszu (odpowiednio wektor wzdłuż ostatniego wymiaru). Zatem,
wartości.kształt = wskaźniki.kształt = kształt.wejściowy[:-1] + [k]
Jeśli dwa elementy są równe, element o niższym indeksie pojawia się jako pierwszy.
Klasy zagnieżdżone
| klasa | TopK.Opcje | Opcjonalne atrybuty dla TopK | |
Stałe
| Strunowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
Metody publiczne
| statyczny <T rozszerza TNumer > TopK <T> | |
| Dane wyjściowe <TInt32> | indeksy () Indeksy „wartości” w ostatnim wymiarze „wejściowego”. |
| statyczne TopK.Opcje | posortowane (posortowane boolowskie) |
| Wyjście <T> | wartości () „k” największych elementów wzdłuż każdego ostatniego przekroju wymiarowego. |
Metody dziedziczone
Stałe
publiczny statyczny końcowy ciąg znaków OP_NAME
Nazwa tej operacji znana silnikowi rdzenia TensorFlow
Metody publiczne
public static TopK <T> utwórz (zakres zakresu , wejście argumentu <T>, argument < TInt32 > k, opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację TopK.
Parametry
| zakres | aktualny zakres |
|---|---|
| wejście | 1-D lub wyższy, z ostatnim wymiarem co najmniej „k”. |
| k | 0-D. Liczba górnych elementów, których należy szukać wzdłuż ostatniego wymiaru (wzdłuż każdego wiersza w przypadku macierzy). |
| opcje | przenosi opcjonalne wartości atrybutów |
Zwroty
- nowa instancja TopK
publiczne statyczne TopK.Options posortowane (posortowane boolowskie)
Parametry
| posortowane | Jeśli ma wartość true, powstałe elementy `k` zostaną posortowane malejąco według wartości. |
|---|
publiczne wartości wyjściowe <T> ()
„k” największych elementów wzdłuż każdego ostatniego przekroju wymiarowego.