Class SpaceToBatchND

• Defined in File space_to_batch_nd.h

Inheritance Relationships

Base Type

• public mindspore::ops::PrimitiveC (Class PrimitiveC)

Class Documentation

class SpaceToBatchND : public mindspore::ops::PrimitiveC

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.

~SpaceToBatchND() = default

Destructor.

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.