zzy34407230
/
mindspore2022

﻿mindspore.ops.Split
====================

.. py:class:: mindspore.ops.Split(axis=0, output_num=1)

    根据指定的轴和分割数量对输入Tensor进行分割。

    `input_x` Tensor将被分割为相同shape的子Tensor，且要求 `input_x.shape(axis)` 可被 `output_num` 整除。

    **参数：**

    - **axis** (int) - 指定分割轴。默认值：0。
    - **output_num** (int) - 指定分割数量。其值为正整数。默认值：1。

    **输入：**

    - **input_x** (Tensor) - Tensor的shape为 :math:`(x_1, x_2, ..., x_R)` 。

    **输出：**

    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。