Trả về các giá trị k tối thiểu/tối đa và chỉ số của chúng của toán hạng đầu vào một cách gần đúng.
Xem https://arxiv.org/abs/2206.14286 để biết chi tiết về thuật toán. Op này hiện chỉ được tối ưu hóa trên TPU.
Các lớp lồng nhau
lớp học | KhoảngTopK.Options | Các thuộc tính tùy chọn cho ApproxTopK |
Phương pháp công cộng
khoảngTopK.Options tĩnh | tổng hợpToTopk (Boolean tổng hợpToTopk) |
tĩnh <T mở rộng Số> KhoảngTopK <T> | tạo ( Phạm vi phạm vi, đầu vào Toán hạng <T>, Long 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 thao tác ApproTopK mới. |
Đầu ra <Số nguyên> | chỉ số () Các chỉ số của `giá trị` dọc theo `reduction_dimension` của toán hạng `input`. |
khoảngTopK.Options tĩnh | isMaxK (Boolean isMaxK) |
khoảngTopK.Options tĩnh | mục tiêu thu hồi (Mục tiêu thu hồi nổi) |
khoảngTopK.Options tĩnh | Giảm kích thước (Giảm kích thước dài) |
khoảngTopK.Options tĩnh | giảmInputSizeOverride (Giảm dàiInputSizeOverride) |
Đầu ra <T> | giá trị () Các giá trị k tối thiểu/tối đa dọc theo `reduction_dimension` của toán hạng `input`. |
Phương pháp kế thừa
Phương pháp công cộng
tĩnh công khai approTopK.Options tổng hợpToTopk (Boolean tổng hợpToTopk)
Thông số
tổng hợpToTopk | Khi đúng, tổng hợp các kết quả gần đúng với top-k. Khi sai, trả về kết quả gần đúng. Số lượng kết quả gần đúng được xác định khi triển khai và lớn hơn bằng `k` được chỉ định. |
---|
tĩnh công khai OVERTopK <T> tạo ( Phạm vi phạm vi , đầu vào Toán hạng <T>, Độ dài 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 thao tác ApproTopK mới.
Thông số
phạm vi | phạm vi hiện tại |
---|---|
đầu vào | Mảng để tìm kiếm. Phải có ít nhất 1-D thuộc loại nổi |
k | Chỉ định số min/max-k. |
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 approTopK
chỉ số đầu ra công khai <Integer> ()
Các chỉ số của `giá trị` dọc theo `reduction_dimension` của toán hạng `input`.
tĩnh công khai KhoảngTopK.Options isMaxK (Boolean isMaxK)
Thông số
isMaxK | Khi đúng, tính max-k; mặt khác tính toán min-k. |
---|
tĩnh công khai OVERTopK.Options thu hồiTarget (Mục tiêu thu hồi nổi)
Thông số
thu hồiMục tiêu | Gọi lại mục tiêu cho phép tính gần đúng. Phạm vi trong (0,1] |
---|
tĩnh công khai approTopK.Options lossDimension (Giảm kích thước dài)
Thông số
giảmKích thước | Kích thước nguyên để tìm kiếm. Mặc định: -1. |
---|
tĩnh công khai KhoảngTopK.Options lossInputSizeOverride (Giảm dàiInputSizeOverride)
Thông số
giảmInputSizeOverride | Khi được đặt thành giá trị dương, nó sẽ ghi đè kích thước được xác định bởi `input[reduction_dim]` để đánh giá việc thu hồi. Tùy chọn này hữu ích khi `đầu vào` đã cho chỉ là tập hợp con của tính toán tổng thể trong SPMD hoặc đường ống phân tán, trong đó kích thước đầu vào thực không thể bị trì hoãn bởi hình dạng `đầu vào`. |
---|
giá trị đầu ra công khai <T> ()
Các giá trị k tối thiểu/tối đa dọc theo `reduction_dimension` của toán hạng `input`. Thứ nguyên giống với toán hạng `input` ngoại trừ `reduction_dimension`: khi `aggregate_to_topk` là đúng, thứ nguyên rút gọn là `k`; mặt khác, nó lớn hơn bằng `k` trong đó kích thước được xác định theo cách triển khai.