dòng chảy căng:: ôi:: 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ơ (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 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). Vì 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.

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 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 .

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
)