Class DepthToSpace

Defined in File depth_to_space.h

Inheritance Relationships

Base Type

public mindspore::ops::BaseOperator

Class Documentation

class DepthToSpace : public mindspore::ops::BaseOperator 

Rearrange blocks of depth data into spatial dimensions. Refer to Python API mindspore.ops.DepthToSpace for more details.

Public Functions

inline DepthToSpace(): Constructor.

void Init(const int64_t block_size, const Format &format = NCHW): Init. Refer to the parameters of Python API mindspore.ops.DepthToSpace 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.