Class BatchToSpace

• Defined in File batch_to_space.h

Inheritance Relationships

Base Type

• public mindspore::ops::BaseOperator

Class Documentation

class BatchToSpace : public mindspore::ops::BaseOperator

Divides batch dimension with blocks and interleaves these blocks back into spatial dimensions. Refer to Python API mindspore.ops.BatchToSpace for more details.

Public Functions

inline BatchToSpace()

Constructor.

void Init(const std::vector<int64_t> &block_size, const std::vector<std::vector<int64_t>> &crops)

Init. Refer to the parameters of Python API mindspore.ops.BatchToSpace for the inputs.

void set_block_size(const std::vector<int64_t> &block_size)

Set block_size.

void set_crops(const std::vector<std::vector<int64_t>> &crops)

Set crops.

std::vector<int64_t> get_block_size() const

Get block_size.

Returns

block_size.

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

Get crops.

Returns

crops.