Class ov::op::v0::SpaceToDepth

class SpaceToDepth : public ov::op::Op

SpaceToDepth permutes input tensor blocks of spatial data into depth dimension.


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

   Output node produces a tensor with shape:
   [N, C * blocksize * blocksize, H / blocksize, W / blocksize]

Public Functions

SpaceToDepth(const Output<Node> &data, const SpaceToDepthMode &mode, std::size_t block_size = 1)

Constructs a SpaceToDepth operation.

  • data – - Node producing the input tensor

  • mode – Specifies how the output depth dimension is gathered from block coordinates and the old depth dimension.

  • block_size – - the size of the block of values to be moved

virtual void validate_and_infer_types() override

Verifies that attributes and inputs are consistent and computes output shapes and element types. Must be implemented by concrete child classes so that it can be run any number of times.

Throws if the node is invalid.

virtual bool has_evaluate() const override

Allows to get information about availability of evaluate method for the current operation.