Class Split
Defined in File split.h
Inheritance Relationships
Base Type
public mindspore::ops::BaseOperator
Class Documentation
-
class Split : public mindspore::ops::BaseOperator
Splits the input tensor into output_num of tensors along the given axis and output numbers. Refer to Python API mindspore.ops.Split for more details.
Public Functions
-
inline Split()
Constructor.
-
void Init(const int64_t axis, const int64_t output_num)
Init. Refer to the parameters of Python API mindspore.ops.Split for the inputs.
-
void set_size_splits(const std::vector<int64_t> &size_splits)
Set size_splits.
-
void set_axis(const int64_t axis)
Set axis.
-
void set_output_num(const int64_t output_num)
Set output_num.
-
std::vector<int64_t> get_size_splits() const
Get size_splits.
- Returns
size_splits.
-
int64_t get_axis() const
Get size_splits.
- Returns
size_splits.
-
int64_t get_output_num() const
Get output_num.
- Returns
output_num.
-
inline Split()