Class SpaceToBatchND

• Defined in File space_to_batch_nd.h

Inheritance Relationships

Base Type

• public mindspore::ops::BaseOperator

Class Documentation

class SpaceToBatchND : public mindspore::ops::BaseOperator

Divides spatial dimensions into blocks and combines the block size with the original batch. Refer to Python API mindspore.ops.SpaceToBatchND for more details.

Public Functions

inline SpaceToBatchND()

Constructor.

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

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

void set_paddings(const std::vector<std::vector<int64_t>> paddings)

Set paddings.

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_paddings() const

Get paddings.

Returns

paddings.