TopK

publiczna klasa końcowa TopK

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>
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.
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

Wartość stała: „TopKV2”

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 wyjście < TInt32 > indeksy ()

Indeksy „wartości” w ostatnim wymiarze „wejściowego”.

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.