dòng chảy căng:: ôi:: BatchToSpace

#include <array_ops.h>

BatchToSpace dành cho tensor 4-D loại T.

Bản tóm tắt

Đây là phiên bản kế thừa của BatchToSpaceND tổng quát hơn.

Sắp xếp lại (hoán vị) dữ liệu từ lô thành khối dữ liệu không gian, sau đó cắt xén. Đây là sự chuyển đổi ngược lại của SpaceToBatch. Cụ thể hơn, op này tạo ra một bản sao của tenxơ đầu vào trong đó các giá trị từ thứ nguyên batch được di chuyển trong các khối không gian sang kích thước heightwidth , sau đó cắt xén dọc theo kích thước heightwidth .

Lập luận:

  • phạm vi: Một đối tượng Phạm vi
  • đầu vào: tensor 4-D có hình dạng [batch*block_size*block_size, height_pad/block_size, width_pad/block_size, depth] . Lưu ý rằng kích thước lô của tensor đầu vào phải chia hết cho block_size * block_size .
  • crop: tensor 2-D của các số nguyên không âm có hình dạng [2, 2] . Nó chỉ định số lượng phần tử cần cắt từ kết quả trung gian theo các chiều không gian như sau:
    crops = [[crop_top, crop_bottom], [crop_left, crop_right]]

Trả về:

  • Output : 4-D với hình dạng [batch, height, width, depth] , trong đó:
    height = height_pad - crop_top - crop_bottom
    width = width_pad - crop_left - crop_right

attr block_size phải lớn hơn một. Nó chỉ ra kích thước khối.

Một số ví dụ:

(1) Đối với đầu vào sau có hình dạng [4, 1, 1, 1] và block_size là 2:

[[[[1]]], [[[2]]], [[[3]]], [[[4]]]]

Tensor đầu ra có dạng [1, 2, 2, 1] và có giá trị:

x = [[[[1], [2]], [[3], [4]]]]

(2) Đối với đầu vào sau có hình dạng [4, 1, 1, 3] và block_size là 2:

[[[[1, 2, 3]]], [[[4, 5, 6]]], [[[7, 8, 9]]], [[[10, 11, 12]]]]

Tensor đầu ra có dạng [1, 2, 2, 3] và có giá trị:

x = [[[[1, 2, 3], [4, 5, 6]],
      [[7, 8, 9], [10, 11, 12]]]]

(3) Đối với đầu vào sau có hình dạng [4, 2, 2, 1] và block_size là 2:

x = [[[[1], [3]], [[9], [11]]],
     [[[2], [4]], [[10], [12]]],
     [[[5], [7]], [[13], [15]]],
     [[[6], [8]], [[14], [16]]]]

Tenxơ đầu ra có dạng [1, 4, 4, 1] và có giá trị:

x = [[[[1],   [2],  [3],  [4]],
     [[5],   [6],  [7],  [8]],
     [[9],  [10], [11],  [12]],
     [[13], [14], [15],  [16]]]]

(4) Đối với đầu vào sau có dạng [8, 1, 2, 1] và block_size là 2:

x = [[[[1], [3]]], [[[9], [11]]], [[[2], [4]]], [[[10], [12]]],
     [[[5], [7]]], [[[13], [15]]], [[[6], [8]]], [[[14], [16]]]]

Tensor đầu ra có dạng [2, 2, 4, 1] và có giá trị:

x = [[[[1], [3]], [[5], [7]]],
     [[[2], [4]], [[10], [12]]],
     [[[5], [7]], [[13], [15]]],
     [[[6], [8]], [[14], [16]]]]

Hàm tạo và hàm hủy

BatchToSpace (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input crops, int64 block_size)

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

BatchToSpace

 BatchToSpace(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input crops,
  int64 block_size
)

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