mindspore.ops.tensor_split
===========================

.. image:: https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/master/resource/_static/logo_source.svg
    :target: https://gitee.com/mindspore/mindspore/blob/master/docs/api/api_python/ops/mindspore.ops.func_tensor_split.rst
    :alt: 查看源文件


.. py:function:: mindspore.ops.tensor_split(input, indices_or_sections, axis=0)

    根据指定索引或份数，将输入tensor拆分成多个子tensor。

    参数：
        - **input** (Tensor) - 输入tensor。
        - **indices_or_sections** (Union[int, tuple(int), list(int)]) - 指定索引或份数。

          - 如果是int类型，输入tensor将被拆分成 `indices_or_sections` 份。

            - 如果 :math:`input.shape[axis]` 能被 `indices_or_sections` 整除，那么子切片为相同大小 :math:`input.shape[axis] / n` 。
            - 如果 :math:`input.shape[axis]` 不能被 `indices_or_sections` 整除，那么前 :math:`input.shape[axis] \bmod n` 个切片的大小为 :math:`input.shape[axis] // n + 1` ，其余切片的大小为 :math:`input.shape[axis] // n` 。

          - 如果是tuple(int)或list(int)类型，则表示索引，输入tensor在索引处被拆分。
        - **axis** (int，可选) - `indices_or_sections` 所在的轴。默认 ``0`` 。

    返回：
        由多个tensor组成的tuple。