openvino.runtime.opset15.depth_to_space

openvino.runtime.opset15.depth_to_space(node: Node, mode: str, block_size: int = 1, name: str | None = None) → Node

Rearranges input tensor from depth into blocks of spatial data.

Values from the height and width dimensions are moved to the depth dimension.

Input tensor has shape [N,C,H,W], where N is the batch axis, C is the channel or depth, H is the height and W is the width.

Output node produces a tensor with shape:

[N, C * block_size * block_size, H / block_size, W / block_size]

Parameters:

• node – The node with input tensor data.

• mode –

  Specifies how the input depth dimension is split to block coordinates

  blocks_first: The input is divided to [block_size, …, block_size, new_depth] depth_first: The input is divided to [new_depth, block_size, …, block_size]

• block_size – The size of the spatial block of values describing how the tensor’s data is to be rearranged.

• name – Optional output node name.

Returns:

The new node performing an DepthToSpace operation on its input tensor.