dòng chảy căng:: ôi:: Phần tử thứ N

#include <nn_ops.h>

Tìm các giá trị của thống kê n cho chiều cuối cùng.

Bản tóm tắt

Nếu đầu vào là một vectơ (xếp hạng-1), hãy tìm các mục có giá trị nhỏ thứ n trong vectơ và xuất giá trị của chúng dưới dạng tenxơ vô hướng.

Đối với ma trận (tương ứng với đầu vào có thứ hạng cao hơn), tính toán các mục nhập có giá trị nhỏ thứ n trong mỗi hàng (vectơ tương ứng dọc theo chiều cuối cùng). Vì vậy, 

values.shape = input.shape[:-1]

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 n+1 .
  • n: 0-D. Vị trí của vectơ đã sắp xếp để chọn dọc theo chiều cuối cùng (dọc theo mỗi hàng đối với ma trận). Phạm vi hợp lệ của n là [0, input.shape[:-1])

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

  • đảo ngược: Khi được đặt thành True, hãy tìm giá trị lớn thứ n trong vectơ và ngược lại.

Trả về:

  • Output : Thống kê n dọc theo mỗi lát chiều cuối cùng.

Hàm tạo và hàm hủy

NthElement (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input n)
NthElement (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input n, const NthElement::Attrs & attrs)

Thuộc tính công khai

operation
values

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

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Các hàm tĩnh công khai

Reverse (bool x)

Cấu trúc

tenorflow:: ops:: NthElement:: Attrs

Trình thiết lập thuộc tính tùy chọn cho NthElement .

Thuộc tính công khai

hoạt động

Operation operation

giá trị

::tensorflow::Output values

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

Phần tử thứ N

 NthElement(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input n
)

Phần tử thứ N

 NthElement(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input n,
  const NthElement::Attrs & attrs
)

nút

::tensorflow::Node * node() const

toán tử::tenorflow::Đầu vào

 operator::tensorflow::Input() const

toán tử::tenorflow::Đầu ra

 operator::tensorflow::Output() const

Các hàm tĩnh công khai

Đảo ngược

Attrs Reverse(
  bool x
)