2.3K Star 8.1K Fork 4.3K

GVPMindSpore / mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.Layout.rst 1.22 KB
一键复制 编辑 原始数据 按行查看 历史
yao_yf 提交于 2024-03-19 10:44 . code_docs_layout_config_verify_r2.3

mindspore.Layout

.. py:class:: mindspore.Layout(device_matrix, alias_name)

    Layout描述了详细的切分信息。

    .. note::
        - 仅在半自动并行或自动并行模式下有效。
        - `device_matrix` 的累乘结果必须等于一个pipeline stage中的设备数。
        - 当Layout来构建切分策略时,每个别名只允许用于一次张量的切分。

    参数:
        - **device_matrix** (tuple) - 描述设备排列的形状,其元素类型为int。
        - **alias_name** (tuple) - device_matrix的每个轴的别名,其元素类型为字符串。

    异常:
        - **TypeError** - `device_matrix` 不是元组类型。
        - **TypeError** - `alias_name` 不是元组类型。
        - **ValueError** - `device_matrix` 长度不等于 `alias_name` 长度。
        - **TypeError** - `device_matrix` 的元素不是int类型。
        - **TypeError** - `alias_name` 的元素不是str类型。
        - **ValueError** - `alias_name` 的元素是一个空的str。
        - **ValueError** - `alias_name` 的元素为“None”。
        - **ValueError** - `alias_name` 包含重复的元素。

    .. py:method:: to_dict

        将Layout转换为词典 。
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
master

搜索帮助