mindspore.parameter_broadcast
======================================

.. 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/mindspore/mindspore.parameter_broadcast.rst
    :alt: 查看源文件


.. py:function:: mindspore.parameter_broadcast(net, layout, cur_rank=0, initial_rank=0)

    在数据并行维度将参数广播给另外的卡。

    .. warning::
        这是一个实验性API，后续可能修改或删除。

    参数：
        - **net** (Cell) - 参数将被广播的网络。
        - **layout** (Dict) - 参数排布字典。 来自 :func:`mindspore.nn.Cell.parameter_layout_dict` 或
          从文件中读取(如: 通过 :func:`mindspore.set_auto_parallel_context` 接口的 `strategy_ckpt_config`
          参数保存的"strategy.ckpt"文件)。key为参数名， value为该参数的layout。
        - **cur_rank** (int，可选) - 当前卡的rank id。默认值: ``0``。
        - **initial_rank** (int，可选) - 当前流水线并行stage起始rank id。默认值: ``0``。

    异常：
        - **ValueError** - `cur_rank` 不是当前卡的rank_id。
        - **ValueError** - `initial_rank` 不是当前pipeline_stage起始的rank_id。
        - **ValueError** - `layout` 中的参数名在 :func:`mindspore.nn.Cell.parameters_dict` 中找不到。