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.

Returns

block_size.

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

Get paddings.

Returns

paddings.