dòng chảy căng:: ôi:: Hợp nhấtResizeAndPadConv2D
#include <nn_ops.h>
Thực hiện thay đổi kích thước và đệm như một bước tiền xử lý trong quá trình tích chập.
Bản tóm tắt
Thông thường, có thể thực hiện các phép biến đổi không gian hiệu quả hơn như một phần của giai đoạn đóng gói của tích chập, vì vậy hoạt động này cho phép triển khai được tối ưu hóa trong đó các giai đoạn này được hợp nhất với nhau. Điều này ngăn cản sự cần thiết 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ố 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à được đặt mặc định theo thứ tự 'NHWC'. 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]
. - kích thước: Tensor int32 1-D gồm 2 phần tử:
new_height, new_width
. Kích thước mới cho hình ảnh. - 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.
Thuộc tính tùy chọn (xem Attrs
):
- thay đổi kích thước_align_corners: Nếu đúng, tâm của 4 pixel góc của tensor đầu vào và đầu ra được căn chỉnh, giữ nguyên các giá trị ở các pixel góc. Mặc định là sai.
Trả về:
-
Output
: Tenxor đầu ra.
Hàm tạo và hàm hủy | |
---|---|
FusedResizeAndPadConv2D (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input size, :: tensorflow::Input paddings, :: tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding) | |
FusedResizeAndPadConv2D (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input size, :: tensorflow::Input paddings, :: tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding, const FusedResizeAndPadConv2D::Attrs & attrs) |
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 |
Các hàm tĩnh công khai | |
---|---|
ResizeAlignCorners (bool x) |
Cấu trúc | |
---|---|
tensorflow:: ops:: FusedResizeAndPadConv2D:: Attrs | Bộ cài đặt thuộc tính tùy chọn cho FusedResizeAndPadConv2D . |
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
Hợp nhấtResizeAndPadConv2D
FusedResizeAndPadConv2D( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input size, ::tensorflow::Input paddings, ::tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding )
Hợp nhấtResizeAndPadConv2D
FusedResizeAndPadConv2D( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input size, ::tensorflow::Input paddings, ::tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding, const FusedResizeAndPadConv2D::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
Thay đổi kích thướcCăn chỉnh các góc
Attrs ResizeAlignCorners( bool x )