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

org.tensorflow.op.RawOp
końcowa wartość logiczna
równa się (obiekt obiektu)
końcowy wew
Operacja
op ()
Zwróć tę jednostkę obliczeniową jako pojedynczą Operation .
ostatni ciąg
wartość logiczna
równa się (Obiekt arg0)
ostatnia klasa<?>
pobierzKlasę ()
wew
hashCode ()
ostateczna pustka
powiadomić ()
ostateczna pustka
powiadom wszystkich ()
Strunowy
doString ()
ostateczna pustka
czekaj (długi arg0, int arg1)
ostateczna pustka
czekaj (długi arg0)
ostateczna pustka
Czekać ()
org.tensorflow.op.Op
abstrakcyjne środowisko wykonania
środowisko ()
Zwróć środowisko wykonawcze, w którym utworzono tę operację.
abstrakcyjna operacja
op ()
Zwróć tę jednostkę obliczeniową jako pojedynczą Operation .

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.