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.