Class DepthToSpace

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.