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.
-
inline DepthToSpace()