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ê thứ 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). Như vậy,
values.shape = input.shape[:-1]
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
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 )