tensorflow::ops::BatchToSpace

#include <array_ops.h>

BatchToSpace dành cho tensor 4 chiều loại T.

Bản tóm tắt

Đây là phiên bản cũ hơn của BatchToSpaceND , một công cụ tổng quát hơn.

Thao tác này sắp xếp lại (hoán vị) dữ liệu từ batch thành các khối dữ liệu không gian, sau đó cắt xén. Đây là phép biến đổi ngược lại của SpaceToBatch. Cụ thể hơn, thao tác này xuất ra một bản sao của tensor đầu vào, trong đó các giá trị từ chiều batch được di chuyển theo các khối không gian đến height và width , sau đó cắt xén theo height và width .

Luận điểm:

  • phạm vi: Một đối tượng Phạm vi
  • Đầu vào: Tensor 4 chiều với 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 batch của tensor đầu vào phải chia hết cho block_size * block_size .
  • crops: Tensor 2 chiều gồm 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 trên các chiều không gian như sau:
    crops = [[crop_top, crop_bottom], [crop_left, crop_right]]

Trả về:

  • Output : Mô hình 4 chiều 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

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

Một vài ví dụ:

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

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

Tensor đầu ra có hình dạng [1, 2, 2, 1] và 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à kích thước khối là 2:

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

Tensor đầu ra có hình dạng [1, 2, 2, 3] và 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à kích thước khối là 2:

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

Tensor đầu ra có hình dạng [1, 4, 4, 1] và 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ó hình dạng [8, 1, 2, 1] và kích thước khối là 2:

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

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

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

Hàm khởi 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 

operator::tensorflow::Input

 operator::tensorflow::Input() const 

operator::tensorflow::Output

 operator::tensorflow::Output() const