openvino.runtime.opset15.space_to_depth

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

Perform SpaceToDepth operation on the input tensor.

SpaceToDepth rearranges blocks of spatial data into depth. The operator :return: a copy of the input tensor where values from the height and width dimensions are moved to the depth dimension.

Parameters:

• data – The node with data tensor.

• mode –

  Specifies how the output depth dimension is gathered from block coordinates.

  blocks_first: The output depth is gathered from [block_size, …, block_size, C] depth_first: The output depth is gathered from [C, block_size, …, block_size]

• block_size – The size of the block of values to be moved. Scalar value.

• name – Optional output node name.

Returns:

The new node performing a SpaceToDepth operation on input tensor.