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