Class Split

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.

返回

size_splits.

int64_t get_axis() const

Get size_splits.

返回

size_splits.

int64_t get_output_num() const

Get output_num.

返回

output_num.