2.4K Star 8.2K Fork 4.4K

GVPMindSpore / mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.nn.GraphCell.rst 1.33 KB
一键复制 编辑 原始数据 按行查看 历史
luojianing 提交于 2023-05-18 13:04 . add ` for values

mindspore.nn.GraphCell

.. py:class:: mindspore.nn.GraphCell(graph, params_init=None, obf_random_seed=None)

    运行从MindIR加载的计算图。

    此功能仍在开发中。目前 `GraphCell` 不支持修改图结构,在导出MindIR时只能使用shape和类型与输入相同的数据。

    参数:
        - **graph** (FuncGraph) - 从MindIR加载的编译图。
        - **params_init** (dict) - 需要在图中初始化的参数。key为参数名称,类型为字符串,value为 Tensor 或 Parameter。如果参数名在图中已经存在,则更新其值;如果不存在,则忽略。默认值: ``None`` 。
        - **obf_random_seed** (Union[int, None]) - 用于动态混淆保护的混淆随机种子。动态混淆是一种模型保护方法,可以参考 :func:`mindspore.obfuscate_model` 。如果导入的 `graph` 是一个经过混淆的模型,那么须提供 `obf_random_seed` 。 `obf_random_seed` 的取值范围是(0, 9223372036854775807]。默认值: ``None`` 。

    异常:
        - **TypeError** - 如果图不是FuncGraph类型。
        - **TypeError** - 如果 `params_init` 不是字典。
        - **TypeError** - 如果 `params_init` 的key不是字符串。
        - **TypeError** - 如果 `params_init` 的value既不是 Tensor也不是Parameter。
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
r2.3

搜索帮助