Trả về các giá trị duy nhất TopK trong mảng theo thứ tự được sắp xếp. Các
thời gian chạy tỷ lệ thuận với tích của K và kích thước đầu vào. Việc sắp xếp toàn bộ mảng sẽ hiệu quả hơn đối với các giá trị K đủ lớn. Thuật toán trung vị có thể nhanh hơn nhưng khó triển khai hiệu quả trong XLA. Nếu có ít hơn K số duy nhất (không phải NAN), kết quả sẽ được đệm bằng âm vô cực. NaN không bao giờ được trả lại. Các số dưới mức bình thường được xóa về 0. Nếu một phần tử xuất hiện ở nhiều chỉ mục thì chỉ số cao nhất sẽ được trả về. Nếu phần tử TopK không bao giờ xuất hiện trong đầu vào do giá trị đệm thì các chỉ số sẽ được đệm bằng giá trị âm. Nếu một giá trị đệm xuất hiện trong đầu vào và cần có phần đệm thì chỉ số cao nhất của giá trị đệm sẽ được trả về. Ngữ nghĩa không giống như kth_order_statistic.
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 khai
| tĩnh TopKUđộc quyền | |
| Đầu ra < TFloat32 > | đỉnh () |
| Đầu ra < TInt32 > | topkChỉ số () |
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
Phương pháp công khai
công khai tĩnh TopKU tạo duy nhất ( Phạm vi phạm vi , Toán hạng < TFloat32 > đầu vào, Long k)
Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác TopKUnique mới.
Thông số
| phạm vi | phạm vi hiện tại |
|---|
Trả lại
- một phiên bản mới của TopKUnique