2022-06-17 14:28:33 +08:00
|
|
|
|
mindspore.ops.split
|
|
|
|
|
|
====================
|
|
|
|
|
|
|
|
|
|
|
|
.. py:function:: mindspore.ops.split(input_x, axis=0, output_num=1)
|
|
|
|
|
|
|
|
|
|
|
|
根据指定的轴和分割数量对输入Tensor进行分割。
|
|
|
|
|
|
|
2022-10-12 17:20:15 +08:00
|
|
|
|
`input_x` Tensor将被分割为相同shape的子Tensor。要求 `input_x.shape(axis)` 可被 `output_num` 整除。
|
2022-06-17 14:28:33 +08:00
|
|
|
|
|
2022-07-26 10:27:56 +08:00
|
|
|
|
参数：
|
|
|
|
|
|
- **input_x** (Tensor) - Tensor的shape为 :math:`(x_1, x_2, ..., x_R)` 。
|
|
|
|
|
|
- **axis** (int) - 指定分割轴。默认值：0。
|
|
|
|
|
|
- **output_num** (int) - 指定分割数量。其值为正整数。默认值：1。
|
|
|
|
|
|
|
|
|
|
|
|
返回：
|
|
|
|
|
|
tuple[Tensor]，每个输出Tensor的shape相同，即 :math:`(y_1, y_2, ..., y_S)` 。数据类型与 `input_x` 的相同。
|
|
|
|
|
|
|
|
|
|
|
|
异常：
|
|
|
|
|
|
- **TypeError** - `axis` 或 `output_num` 不是int。
|
|
|
|
|
|
- **ValueError** - `axis` 超出[-len(`input_x.shape`), len(`input_x.shape`))范围。或 `output_num` 小于或等于0。
|
|
|
|
|
|
- **ValueError** - `input_x.shape(axis)` 不可被 `output_num` 整除。
|
