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` 整除。