dòng chảy căng:: ôi:: TopK

#include <nn_ops.h>

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

Bản tóm tắt

Nếu đầu vào là một vectơ (cấp 1), hãy tìm k mục lớn nhất trong vectơ và xuất ra 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ó 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 k mục hàng đầu trong mỗi hàng (vectơ tương ứng dọc theo chiều cuối cùng). Như vậy,

values.shape = indices.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.

Lập 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 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).

Thuộc tính tùy chọn (xem Attrs ):

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

Trả về:

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

Hàm tạo và hàm 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 hàm tĩnh công khai

Sorted (bool x)

Cấu trúc

tenorflow:: ops:: TopK:: Attrs

Bộ cài đặt 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 hàm tĩnh công khai

Đã sắp xếp

Attrs Sorted(
  bool x
)