dòng chảy :: hoạt động :: FusedPadConv2D
#include <nn_ops.h>
Thực hiện đệm dưới dạng tiền xử lý trong quá trình tích chập.
Tóm lược
Tương tự như FusedResizeAndPadConv2d, op này cho phép thực hiện tối ưu hóa trong đó giai đoạn chuyển đổi đệ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 để thay đổi kích thước. Việc hợp nhất phần đệm ngăn chặn nhu cầu viết ra các kết quả trung gian dưới dạng toàn bộ, giảm áp lực bộ nhớ và chúng ta có thể nhận được một số tăng độ 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à đơn hàng 'NHWC' được sử dụng thay thế. Bên trong op này sử dụng một bộ đệm cào cho mỗi đồ thị, có nghĩa là nó sẽ chặn nếu nhiều phiên bản đang được chạy song song. Điều này là do toán tử này chủ yếu là tối ưu hóa để giảm thiểu việc sử dụng bộ nhớ.
Tranh luận:
- phạm vi: Một đối tượng Phạm vi
- đầu vào: 4-D với shape
[batch, in_height, in_width, in_channels]
. - paddings: Ma trận hai cột xác định kích thước phần đệm. Số hàng phải giống với thứ hạng của
input
. - filter: 4-D với shape
[filter_height, filter_width, in_channels, out_channels]
. - sải bước: 1-D chiều dài 4. Sải chân của cửa sổ trượt cho mỗi kích thước
input
. Phải theo thứ tự như thứ nguyên được chỉ định với định dạng. - padding: Loại thuật toán đệm sẽ sử dụng.
Lợi nhuận:
-
Output
: Bộ căng đầu ra.
Người xây dựng và Người phá 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ử :: tensorflow :: Đầu vào
operator::tensorflow::Input() const
toán tử :: tensorflow :: Đầu ra
operator::tensorflow::Output() const