mindspore/docs/api/api_python/nn/mindspore.nn.GraphCell.rst

20 lines
1.3 KiB
ReStructuredText
Raw Normal View History

2022-02-14 11:35:00 +08:00
mindspore.nn.GraphCell
======================
.. py:class:: mindspore.nn.GraphCell(graph, params_init=None, obf_password=None)
2022-02-14 11:35:00 +08:00
运行从MindIR加载的计算图。
此功能仍在开发中。目前 `GraphCell` 不支持修改图结构在导出MindIR时只能使用shape和类型与输入相同的数据。
2022-07-22 16:25:38 +08:00
参数:
2022-09-01 17:02:07 +08:00
- **graph** (FuncGraph) - 从MindIR加载的编译图。
2022-07-22 16:25:38 +08:00
- **params_init** (dict) - 需要在图中初始化的参数。key为参数名称类型为字符串value为 Tensor 或 Parameter。如果参数名在图中已经存在则更新其值如果不存在则忽略。默认值None。
- **obf_password** (int) - 用于动态混淆保护的password。动态混淆是一种模型保护方法可以参考 :func:`mindspore.train.serialization.obfuscate_model` 。如果导入的 `graph` 是一个经过混淆的模型,那么 `obf_password` 应该要提供。 `obf_password` 的取值范围是(0, 9223372036854775807]。默认值None。
2022-02-14 11:35:00 +08:00
2022-07-22 16:25:38 +08:00
异常:
- **TypeError** - 如果图不是FuncGraph类型。
- **TypeError** - 如果 `params_init` 不是字典。
- **TypeError** - 如果 `params_init` 的key不是字符串。
- **TypeError** - 如果 `params_init` 的value既不是 Tensor也不是Parameter。