Class BatchToSpaceND

• Defined in File batch_to_space_nd.h

Inheritance Relationships

Base Type

• public mindspore::ops::BaseOperator

Class Documentation

class BatchToSpaceND : public mindspore::ops::BaseOperator

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

Public Functions

inline BatchToSpaceND()

Constructor.

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

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

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

Set crops.

void set_block_shape(std::vector<int64_t> block_shape)

Set block_shape.

std::vector<int64_t> get_block_shape() const

Get block_shape.

Returns

block_shape.

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

Get crops.

Returns

crops.