Google I / O hoạt động trở lại từ ngày 18 đến 20 tháng 5! Đặt chỗ và xây dựng lịch trình của bạn Đăng ký ngay

dòng chảy :: hoạt động :: TopK

#include <nn_ops.h>

Tìm giá trị và chỉ số của k phần tử lớn nhất cho thứ nguyên cuối cùng.

Tóm lược

Nếu đầu vào là một vectơ (hạng-1), hãy tìm k mục lớn nhất trong vectơ và xuất giá trị và chỉ số của chúng dưới dạng vectơ. Do đó, values[j] là mục nhập lớn nhất thứ j trong input và chỉ số của nó là indices[j] .

Đối với ma trận (tương ứng với đầu vào xếp hạng cao hơn), tính k mục hàng đầu trong mỗi hàng (véc tơ tương ứng dọc theo thứ nguyên cuối cùng). Như vậy,

values.shape = indices.shape = input.shape[:-1] + [k]

Nếu hai phần tử bằng nhau, phần tử có chỉ số thấp hơn sẽ xuất hiện đầu tiên.

Tranh luận:

  • phạm vi: Một đối tượng Phạm vi
  • đầu vào: 1-D hoặc cao hơn với kích thước cuối cùng ít nhất là k .
  • k: 0-D. Số phần tử hàng đầu cần tìm dọc theo thứ nguyên cuối cùng (dọc theo mỗi hàng đối với ma trận).

Các thuộc tính tùy chọn (xem Phần Attrs ):

  • đã sắp xếp: Nếu đúng, k phần tử thu được sẽ được sắp xếp theo các giá trị theo thứ tự giảm dần.

Lợi nhuận:

  • Giá trị Output : k phần tử lớn nhất dọc theo mỗi lát cắt chiều cuối cùng.
  • Chỉ số Output : Chỉ số của các values trong kích thước cuối cùng của input .

Người xây dựng và Người phá hủy

TopK (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input k)
TopK (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input k, const TopK::Attrs & attrs)

Thuộc tính công khai

indices
operation
values

Các chức năng tĩnh công khai

Sorted (bool x)

Cấu trúc

tensorflow :: ops :: TopK :: Attrs

Bộ thiết lập thuộc tính tùy chọn cho TopK .

Thuộc tính công khai

chỉ số

::tensorflow::Output indices

hoạt động

Operation operation

giá trị

::tensorflow::Output values

Chức năng công cộng

TopK

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k
)

TopK

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k,
  const TopK::Attrs & attrs
)

Các chức năng tĩnh công khai

Đã sắp xếp

Attrs Sorted(
  bool x
)