tensorflow::ops::TopK

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ơ (hạng 1), hàm sẽ tìm k phần tử lớn nhất trong vectơ đó và xuất ra giá trị và chỉ số của chúng dưới dạng vectơ. Ví dụ: values[j] là phần tử lớn 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 có hạng cao hơn), hàm này tính toán k phần tử hàng đầu trong mỗi hàng (tương ứng với vectơ dọc theo chiều cuối cùng). Do đó,

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 trước.

Luận điểm:

• phạm vi: Một đối tượng Phạm vi
• Đầu vào: 1 chiều trở lên với chiều cuối cùng ít nhất là k .
• k: 0-D. Số lượng phần tử hàng đầu 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).

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

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

Trả về:

• 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 values trong chiều cuối cùng của input .

Thuộc tính công khai

::tensorflow::Output indices

Operation operation

::tensorflow::Output values

 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
)

Attrs Sorted(
  bool x
)