forked from mindspore-Ecosystem/mindspore
26 lines
984 B
ReStructuredText
26 lines
984 B
ReStructuredText
|
mindspore.ops.split
|
|||
|
|
====================
|
|||
|
|
|
|||
|
|
.. py:function:: mindspore.ops.split(input_x, axis=0, output_num=1)
|
|||
|
|
|
|||
|
|
根据指定的轴和分割数量对输入Tensor进行分割。
|
|||
|
|
|
|||
|
|
`input_x` Tensor将被分割为相同shape的子Tensor，且要求 `input_x.shape(axis)` 可被 `output_num` 整除。
|
|||
|
|
|
|||
|
|
**参数：**
|
|||
|
|
|
|||
|
|
- **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` 整除。
|