tensorflow::ops::NthElement

#include <nn_ops.h>

Tìm giá trị của thống kê thứ tự n 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 các phần tử có giá trị nhỏ thứ n trong vectơ đó và xuất ra giá trị của chúng dưới dạng tensor vô hướng.

Đố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 các phần tử là giá trị nhỏ thứ n trong mỗi hàng (tương ứng với vectơ dọc theo chiều cuối cùng). Do đó,

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

Luận điểm:

  • phạm vi: Một đối tượng Phạm vi
  • Đầu vào: Không gian 1 chiều trở lên với chiều cuối cùng ít nhất là n+1 .
  • n: 0-D. Vị trí của vectơ đã được 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])

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

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

Trả về:

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

Hàm khởi 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

tensorflow::ops::NthElement::Attrs

Các phương thức 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

NthElement

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

NthElement

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

nút

::tensorflow::Node * node() const 

operator::tensorflow::Input

 operator::tensorflow::Input() const 

operator::tensorflow::Output

 operator::tensorflow::Output() const 

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

Đảo ngược

Attrs Reverse(
  bool x
)