tensorflow::ops::FusedPadConv2D
#include <nn_ops.h>Thực hiện thao tác đệm như một bước tiền xử lý trong quá trình tích chập.
Bản tóm tắt
Tương tự như FusedResizeAndPadConv2d, toán tử này cho phép triển khai tối ưu hóa bằng cách kết hợp giai đoạn biến đổi đệm không gian với việc 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 đệm giúp tránh việc phải 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ố lợi ích về độ trễ bằng cách hợp nhất các phép tính biến đổi. Thuộc tính data_format cho Conv2D không được toán tử này hỗ trợ, thay vào đó sử dụng thứ tự 'NHWC'. Bên trong, toán tử này sử dụng một bộ đệm tạm thời duy nhất cho mỗi đồ thị, điều này có nghĩa là nó sẽ bị chặn nếu nhiều phiên bản được chạy song song. Điều này là do toán tử này chủ yếu là một tối ưu hóa để giảm thiểu việc sử dụng bộ nhớ.
Luận điểm:
- phạm vi: Một đối tượng Phạm vi
- Đầu vào: 4 chiều với hình dạng
[batch, in_height, in_width, in_channels]. - paddings: Một ma trận hai cột chỉ định kích thước phần đệm. Số hàng phải bằng với hạng của ma
input. - bộ lọc: 4 chiều với hình dạng
[filter_height, filter_width, in_channels, out_channels]. - Bước nhảy: 1 chiều có độ dài 4. Bước nhảy của cửa sổ trượt cho mỗi chiều của
input. Phải theo cùng thứ tự với chiều được chỉ định bằng định dạng. - padding: Loại thuật toán đệm cần sử dụng.
Trả về:
-
Output: Tensor đầu ra.
Hàm khởi 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
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const