dòng chảy căng:: ôi:: MaxPoolWithArgmax
#include <nn_ops.h>
Thực hiện gộp tối đa trên đầu vào và đầu ra cả giá trị và chỉ số tối đa.
Bản tóm tắt
Các chỉ số trong argmax
được làm phẳng, sao cho giá trị tối đa tại vị trí [b, y, x, c]
trở thành chỉ mục được làm phẳng: (y * width + x) * channels + c
nếu include_batch_in_index
là Sai; ((b * height + y) * width + x) * channels + c
nếu include_batch_in_index
là Đúng.
Các chỉ số được trả về luôn ở dạng [0, height) x [0, width)
trước khi làm phẳng, ngay cả khi có liên quan đến phần đệm và câu trả lời đúng về mặt toán học nằm ở bên ngoài (âm hoặc quá lớn). Đây là một lỗi nhưng việc khắc phục nó rất khó thực hiện theo cách tương thích ngược an toàn, đặc biệt là do bị làm phẳng.
Lập luận:
- phạm vi: Một đối tượng Phạm vi
- đầu vào: 4-D với hình dạng
[batch, height, width, channels]
. Đầu vào để gộp lại. - ksize: Kích thước của cửa sổ cho từng chiều của tensor đầu vào.
- bước tiến: Bước tiến của cửa sổ trượt cho từng chiều của tensor đầu vào.
- phần đệm: Loại thuật toán đệm sẽ sử dụng.
Thuộc tính tùy chọn (xem Attrs
):
- include_batch_in_index: Có bao gồm thứ nguyên lô trong chỉ mục phẳng của
argmax
hay không.
Trả về:
- Đầu ra
Output
ra: Tenor đầu ra gộp tối đa. -
Output
argmax: 4-D. Các chỉ số làm phẳng của các giá trị tối đa được chọn cho mỗi đầu ra.
Hàm tạo và hàm hủy | |
---|---|
MaxPoolWithArgmax (const :: tensorflow::Scope & scope, :: tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding) | |
MaxPoolWithArgmax (const :: tensorflow::Scope & scope, :: tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding, const MaxPoolWithArgmax::Attrs & attrs) |
Thuộc tính công khai | |
---|---|
argmax | |
operation | |
output |
Các hàm tĩnh công khai | |
---|---|
IncludeBatchInIndex (bool x) | |
Targmax (DataType x) |
Cấu trúc | |
---|---|
tensorflow:: ops:: MaxPoolWithArgmax:: Attrs | Trình thiết lập thuộc tính tùy chọn cho MaxPoolWithArgmax . |
Thuộc tính công khai
argmax
::tensorflow::Output argmax
hoạt động
Operation operation
đầu ra
::tensorflow::Output output
Chức năng công cộng
MaxPoolWithArgmax
MaxPoolWithArgmax( const ::tensorflow::Scope & scope, ::tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding )
MaxPoolWithArgmax
MaxPoolWithArgmax( const ::tensorflow::Scope & scope, ::tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding, const MaxPoolWithArgmax::Attrs & attrs )
Các hàm tĩnh công khai
Bao gồmBatchInIndex
Attrs IncludeBatchInIndex( bool x )
Targmax
Attrs Targmax( DataType x )