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.
-
inline BatchToSpace()