ngraph.opset3.space_to_batch

ngraph.opset3.space_to_batch(data: Union[_pyngraph.Node, int, float, numpy.ndarray], block_shape: Union[_pyngraph.Node, int, float, numpy.ndarray], pads_begin: Union[_pyngraph.Node, int, float, numpy.ndarray], pads_end: Union[_pyngraph.Node, int, float, numpy.ndarray], name: Optional[str] = None)_pyngraph.Node

Perform SpaceToBatch operation on the input tensor.

SpaceToBatch permutes data tensor blocks of spatial data into batch dimension. The operator returns a copy of the input tensor where values from spatial blocks dimensions are moved in the batch dimension

Parameters
  • data – Node producing the data tensor.

  • block_shape – The sizes of the block of values to be moved.

  • pads_begin – Specifies the padding for the beginning along each axis of data.

  • pads_end – Specifies the padding for the ending along each axis of data.

  • name – Optional output node name.

Returns

The new node performing a SpaceToBatch operation.