Versioned name: SpaceToDepth-1
Category: Data movement
Short description: SpaceToDepth operation rearranges data from the spatial dimensions of the input tensor into depth dimension of the output tensor.
Attributes
block_size ^ (len(input.shape) - 2)
.int
[block_size, ..., block_size, C]
[C, block_size, ..., block_size]
string
Inputs
data
- input tensor of any type with rank >= 3. Required.Outputs
[N, C * (block_size ^ K), D1 / block_size, D2 / block_size, ..., DK / block_size]
.Detailed description
SpaceToDepth operation permutes element from the input tensor with shape [N, C, D1, D2, ..., DK]
, to the output tensor where values from the input spatial dimensions D1, D2, ..., DK
are moved to the new depth dimension. Refer to the ONNX* specification for an example of the 4D input tensor case.
The operation is equivalent to the following transformation of the input tensor data
with K
spatial dimensions of shape [N, C, D1, D2, ..., DK]
to Y output tensor. If mode = blocks_first
:
x' = reshape(data, [N, C, D1/block_size, block_size, D2/block_size, block_size, ... , DK/block_size, block_size]) x'' = transpose(x', [0, 3, 5, ..., K + (K + 1), 1, 2, 4, ..., K + K]) y = reshape(x'', [N, C * (block_size ^ K), D1 / block_size, D2 / block_size, ... , DK / block_size])
If mode = depth_first
:
x' = reshape(data, [N, C, D1/block_size, block_size, D2/block_size, block_size, ..., DK/block_size, block_size]) x'' = transpose(x', [0, 1, 3, 5, ..., K + (K + 1), 2, 4, ..., K + K]) y = reshape(x'', [N, C * (block_size ^ K), D1 / block_size, D2 / block_size, ..., DK / block_size])
Example