TopK
Sử dụng bộ sưu tập để sắp xếp ngăn nắp các trang
Lưu và phân loại nội dung dựa trên lựa chọn ưu tiên của bạn.
Tìm các giá trị và chỉ số của phần tử lớn nhất `k` cho chiều cuối cùng.
Nếu đầu vào là một vectơ (xếp hạng-1), hãy tìm các mục lớn nhất `k` trong vectơ và xuất các giá trị cũng như chỉ số của chúng dưới dạng vectơ. Do đó, `values[j]` là mục lớn nhất thứ `j` trong `input` và chỉ mục của nó là `indices[j]`.
Đối với ma trận (tương ứng với đầu vào có thứ hạng cao hơn), hãy tính các mục `k` trên cùng trong mỗi hàng (vectơ tương ứng dọc theo chiều cuối cùng). Như vậy,
value.shape = chỉ số.shape = input.shape[:-1] + [k]
Nếu hai phần tử bằng nhau thì phần tử có chỉ số thấp hơn sẽ xuất hiện đầu tiên.
Hằng số
| Sợi dây | OP_NAME | Tên của op này, được biết đến bởi công cụ lõi TensorFlow |
Phương pháp kế thừa
Từ lớp java.lang.Object | boolean | bằng (Đối tượng arg0) |
| Lớp cuối cùng<?> | getClass () |
| int | Mã Băm () |
| khoảng trống cuối cùng | thông báo () |
| khoảng trống cuối cùng | thông báoTất cả () |
| Sợi dây | toString () |
| khoảng trống cuối cùng | chờ đã (arg0 dài, int arg1) |
| khoảng trống cuối cùng | chờ đã (arg0 dài) |
| khoảng trống cuối cùng | Chờ đợi () |
Hằng số
Chuỗi cuối cùng tĩnh công khai OP_NAME
Tên của op này, được biết đến bởi công cụ lõi TensorFlow
Giá trị không đổi: "TopKV2"
Phương pháp công cộng
Phương thức xuất xưởng để tạo một lớp bao bọc một hoạt động TopK mới.
Thông số
| phạm vi | phạm vi hiện tại |
|---|
| đầu vào | 1-D trở lên với chiều cuối cùng ít nhất là `k`. |
|---|
| k | 0-D. Số phần tử trên cùng cần tìm dọc theo chiều cuối cùng (dọc theo mỗi hàng đối với ma trận). |
|---|
| tùy chọn | mang các giá trị thuộc tính tùy chọn |
|---|
Trả lại
- một phiên bản mới của TopK
Đầu ra công khai < TInt32 > chỉ số ()
Các chỉ số của `giá trị` trong chiều cuối cùng của `đầu vào`.
TopK.Options tĩnh công cộng được sắp xếp (sắp xếp Boolean)
Thông số
| sắp xếp | Nếu đúng thì các phần tử `k` thu được sẽ được sắp xếp theo các giá trị theo thứ tự giảm dần. |
|---|
giá trị đầu ra công khai <T> ()
Phần tử lớn nhất `k` dọc theo mỗi lát chiều cuối cùng.
Trừ phi có lưu ý khác, nội dung của trang này được cấp phép theo Giấy phép ghi nhận tác giả 4.0 của Creative Commons và các mẫu mã lập trình được cấp phép theo Giấy phép Apache 2.0. Để biết thông tin chi tiết, vui lòng tham khảo Chính sách trang web của Google Developers. Java là nhãn hiệu đã đăng ký của Oracle và/hoặc các đơn vị liên kết với Oracle.
Cập nhật lần gần đây nhất: 2025-07-26 UTC.
[null,null,["Cập nhật lần gần đây nhất: 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."]]
