dòng chảy căng:: ôi:: FusedPadConv2D
#include <nn_ops.h>
Thực hiện phần đệm như một phần tiền xử lý trong quá trình tích chập.
Bản tóm tắt
Tương tự như FusedResizeAndPadConv2d, op này cho phép triển khai được tối ưu hóa trong đó giai đoạn chuyển đổi phần đệm không gian được hợp nhất với tra cứu im2col, nhưng trong trường hợp này không cần lọc song tuyến tính để thay đổi kích thước. Việc kết hợp phần đệm giúp ngăn chặn nhu cầu ghi ra các kết quả trung gian dưới dạng toàn bộ tensor, giảm áp lực bộ nhớ và chúng ta có thể đạt được một số mức tăng về độ trễ bằng cách hợp nhất các phép tính chuyển đổi. Thuộc tính data_format cho Conv2D không được op này hỗ trợ và thay vào đó, thứ tự 'NHWC' được sử dụng. Trong nội bộ, op này sử dụng một bộ đệm đầu cho mỗi biểu đồ, có nghĩa là nó sẽ chặn nếu nhiều phiên bản đang chạy song song. Điều này là do toán tử này chủ yếu là sự tối ưu hóa để giảm thiểu việc sử dụng bộ nhớ.
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, in_height, in_width, in_channels]
. - phần đệm: Ma trận hai cột chỉ định kích thước phần đệm. Số lượng hàng phải giống với thứ hạng của
input
. - bộ lọc: 4-D có hình dạng
[filter_height, filter_width, in_channels, out_channels]
. - bước tiến: 1-D có chiều dài 4. Bước tiến của cửa sổ trượt cho từng chiều của
input
. Phải theo cùng thứ tự với kích thước được chỉ định bằng định dạng. - phần đệm: Loại thuật toán đệm sẽ sử dụng.
Trả về:
-
Output
: Tenxor đầu ra.
Hàm tạo và hàm hủy | |
---|---|
FusedPadConv2D (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input paddings, :: tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding) |
Thuộc tính công khai | |
---|---|
operation | |
output |
Chức năng công cộng | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Thuộc tính công khai
hoạt động
Operation operation
đầu ra
::tensorflow::Output output
Chức năng công cộng
FusedPadConv2D
FusedPadConv2D( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input paddings, ::tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding )
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