Class SpaceToBatch

Inheritance Relationships

Base Type

Class Documentation

class SpaceToBatch : public mindspore::ops::PrimitiveC

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.

~SpaceToBatch() = default

Destructor.

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.

Returns

block_size.

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

Get paddings.

Returns

paddings.