TopK
Zadbaj o dobrą organizację dzięki kolekcji
Zapisuj i kategoryzuj treści zgodnie ze swoimi preferencjami.
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.
Stałe
Strunowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
Metody dziedziczone
Z klasy java.lang.Object 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ć () |
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
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 |
---|
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.
O ile nie stwierdzono inaczej, treść tej strony jest objęta licencją Creative Commons – uznanie autorstwa 4.0, a fragmenty kodu są dostępne na licencji Apache 2.0. Szczegółowe informacje na ten temat zawierają zasady dotyczące witryny Google Developers. Java jest zastrzeżonym znakiem towarowym firmy Oracle i jej podmiotów stowarzyszonych.
Ostatnia aktualizacja: 2025-07-26 UTC.
[null,null,["Ostatnia aktualizacja: 2025-07-26 UTC."],[],[],null,["# TopK\n\npublic final class **TopK** \nFinds values and indices of the \\`k\\` largest elements for the last dimension.\n\n\nIf the input is a vector (rank-1), finds the \\`k\\` largest entries in the vector\nand outputs their values and indices as vectors. Thus \\`values\\[j\\]\\` is the\n\\`j\\`-th largest entry in \\`input\\`, and its index is \\`indices\\[j\\]\\`.\n\n\nFor matrices (resp. higher rank input), computes the top \\`k\\` entries in each\nrow (resp. vector along the last dimension). Thus,\n\n\nvalues.shape = indices.shape = input.shape\\[:-1\\] + \\[k\\]\n\n\nIf two elements are equal, the lower-index element appears first.\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n### Nested Classes\n\n|-------|---|---|------------------------------------------------------------------------------|\n| class | [TopK.Options](/jvm/api_docs/java/org/tensorflow/op/nn/TopK.Options) || Optional attributes for [TopK](/jvm/api_docs/java/org/tensorflow/op/nn/TopK) |\n\n### Constants\n\n|--------|-----------------------------------------------------------------|---------------------------------------------------------|\n| String | [OP_NAME](/jvm/api_docs/java/org/tensorflow/op/nn/TopK#OP_NAME) | The name of this op, as known by TensorFlow core engine |\n\n### Public Methods\n\n|--------------------------------------------------------------------------------------------------------------------------------------------------|----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| static \\\u003cT extends [TNumber](/jvm/api_docs/java/org/tensorflow/types/family/TNumber)\\\u003e [TopK](/jvm/api_docs/java/org/tensorflow/op/nn/TopK)\\\u003cT\\\u003e | [create](/jvm/api_docs/java/org/tensorflow/op/nn/TopK#create(org.tensorflow.op.Scope, org.tensorflow.Operand\u003cT\u003e, org.tensorflow.Operand\u003corg.tensorflow.types.TInt32\u003e, org.tensorflow.op.nn.TopK.Options...))([Scope](/jvm/api_docs/java/org/tensorflow/op/Scope) scope, [Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003cT\\\u003e input, [Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TInt32](/jvm/api_docs/java/org/tensorflow/types/TInt32)\\\u003e k, [Options...](/jvm/api_docs/java/org/tensorflow/op/nn/TopK.Options) options) Factory method to create a class wrapping a new TopK operation. |\n| [Output](/jvm/api_docs/java/org/tensorflow/Output)\\\u003c[TInt32](/jvm/api_docs/java/org/tensorflow/types/TInt32)\\\u003e | [indices](/jvm/api_docs/java/org/tensorflow/op/nn/TopK#indices())() The indices of \\`values\\` within the last dimension of \\`input\\`. |\n| static [TopK.Options](/jvm/api_docs/java/org/tensorflow/op/nn/TopK.Options) | [sorted](/jvm/api_docs/java/org/tensorflow/op/nn/TopK#sorted(java.lang.Boolean))(Boolean sorted) |\n| [Output](/jvm/api_docs/java/org/tensorflow/Output)\\\u003cT\\\u003e | [values](/jvm/api_docs/java/org/tensorflow/op/nn/TopK#values())() The \\`k\\` largest elements along each last dimensional slice. |\n\n### Inherited Methods\n\nFrom class [org.tensorflow.op.RawOp](/jvm/api_docs/java/org/tensorflow/op/RawOp) \n\n|----------------------------------------------------------|---------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| final boolean | [equals](/jvm/api_docs/java/org/tensorflow/op/RawOp#equals(java.lang.Object))(Object obj) |\n| final int | [hashCode](/jvm/api_docs/java/org/tensorflow/op/RawOp#hashCode())() |\n| [Operation](/jvm/api_docs/java/org/tensorflow/Operation) | [op](/jvm/api_docs/java/org/tensorflow/op/RawOp#op())() Return this unit of computation as a single [Operation](/jvm/api_docs/java/org/tensorflow/Operation). |\n| final String | [toString](/jvm/api_docs/java/org/tensorflow/op/RawOp#toString())() |\n\nFrom class java.lang.Object \n\n|------------------|---------------------------|\n| boolean | equals(Object arg0) |\n| final Class\\\u003c?\\\u003e | getClass() |\n| int | hashCode() |\n| final void | notify() |\n| final void | notifyAll() |\n| String | toString() |\n| final void | wait(long arg0, int arg1) |\n| final void | wait(long arg0) |\n| final void | wait() |\n\nFrom interface [org.tensorflow.op.Op](/jvm/api_docs/java/org/tensorflow/op/Op) \n\n|-----------------------------------------------------------------------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| abstract [ExecutionEnvironment](/jvm/api_docs/java/org/tensorflow/ExecutionEnvironment) | [env](/jvm/api_docs/java/org/tensorflow/op/Op#env())() Return the execution environment this op was created in. |\n| abstract [Operation](/jvm/api_docs/java/org/tensorflow/Operation) | [op](/jvm/api_docs/java/org/tensorflow/op/Op#op())() Return this unit of computation as a single [Operation](/jvm/api_docs/java/org/tensorflow/Operation). |\n\nConstants\n---------\n\n#### public static final String\n**OP_NAME**\n\nThe name of this op, as known by TensorFlow core engine \nConstant Value: \"TopKV2\"\n\nPublic Methods\n--------------\n\n#### public static [TopK](/jvm/api_docs/java/org/tensorflow/op/nn/TopK)\\\u003cT\\\u003e\n**create**\n([Scope](/jvm/api_docs/java/org/tensorflow/op/Scope) scope, [Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003cT\\\u003e input, [Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TInt32](/jvm/api_docs/java/org/tensorflow/types/TInt32)\\\u003e k, [Options...](/jvm/api_docs/java/org/tensorflow/op/nn/TopK.Options) options)\n\nFactory method to create a class wrapping a new TopK operation. \n\n##### Parameters\n\n| scope | current scope |\n| input | 1-D or higher with last dimension at least \\`k\\`. |\n| k | 0-D. Number of top elements to look for along the last dimension (along each row for matrices). |\n| options | carries optional attributes values |\n|---------|-------------------------------------------------------------------------------------------------|\n\n##### Returns\n\n- a new instance of TopK \n\n#### public [Output](/jvm/api_docs/java/org/tensorflow/Output)\\\u003c[TInt32](/jvm/api_docs/java/org/tensorflow/types/TInt32)\\\u003e\n**indices**\n()\n\nThe indices of \\`values\\` within the last dimension of \\`input\\`. \n\n#### public static [TopK.Options](/jvm/api_docs/java/org/tensorflow/op/nn/TopK.Options)\n**sorted**\n(Boolean sorted)\n\n\u003cbr /\u003e\n\n##### Parameters\n\n| sorted | If true the resulting \\`k\\` elements will be sorted by the values in descending order. |\n|--------|----------------------------------------------------------------------------------------|\n\n#### public [Output](/jvm/api_docs/java/org/tensorflow/Output)\\\u003cT\\\u003e\n**values**\n()\n\nThe \\`k\\` largest elements along each last dimensional slice."]]