lớp chung kết công khai TopK
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.
Các lớp lồng nhau
| lớp học | TopK.Options | Thuộc tính tùy chọn cho TopK | |
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 công cộng
| tĩnh <T mở rộng TNumber > TopK <T> | |
| Đầu ra < 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 sắp xếp (sắp xếp Boolean) |
| Đầu ra <T> | giá trị () Phần tử lớn nhất `k` dọc theo mỗi lát chiều cuối cùng. |
Phương pháp kế thừa
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
công khai tĩnh TopK <T> tạo ( Phạm vi phạm vi , đầu vào Toán hạng <T>, Toán hạng < TInt32 > k, Tùy chọn... tùy chọn)
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
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. |
|---|