Class SpaceToBatch

Defined in File space_to_batch.h

Inheritance Relationships

Base Type

public mindspore::ops::BaseOperator

Class Documentation

class SpaceToBatch : public mindspore::ops::BaseOperator 

Divides spatial dimensions into blocks and combines the block size with the original batch. Refer to Python API mindspore.ops.SpaceToBatch for more details.

Public Functions

inline SpaceToBatch(): Constructor.

void Init(const std::vector<int64_t> block_size, const std::vector<std::vector<int64_t>> &paddings): Init. Refer to the parameters of python API mindspore.ops.SpaceToBatch for the inputs.

void set_paddings(const std::vector<std::vector<int64_t>> &paddings): Set paddings.

void set_block_size(const std::vector<int64_t> block_size): Set block_size.

std::vector<int64_t> get_block_size() const

Get block_size.

返回: block_size.

std::vector<std::vector<int64_t>> get_paddings() const

Get paddings.

返回: paddings.