Class SpaceToDepth

Defined in File space_to_depth.h

Inheritance Relationships

Base Type

public mindspore::ops::BaseOperator

Class Documentation

class SpaceToDepth : public mindspore::ops::BaseOperator 

Rearranges blocks of spatial data into depth. Refer to Python API mindspore.ops.SpaceToDepth for more details.

Public Functions

inline SpaceToDepth(): Constructor.

void Init(const int64_t block_size, const Format &format = NCHW): Init. Refer to the parameters of Python API mindspore.ops.SpaceToDepth for the inputs.

void set_block_size(const int64_t block_size): Set block_size.

int64_t get_block_size() const

Get block_size.

Returns: block_size.

void set_format(const Format &format): Set format.

Format get_format() const

Get format.

Returns: format.