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ề:

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
)