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

19 lines
951 B
ReStructuredText
Raw Normal View History

2022-02-14 11:35:00 +08:00
mindspore.nn.GraphCell
======================
2022-02-25 17:26:18 +08:00
.. py:class:: mindspore.nn.GraphCell(graph, params_init=None)
2022-02-14 11:35:00 +08:00
运行从MindIR加载的计算图。
此功能仍在开发中。目前 `GraphCell` 不支持修改图结构在导出MindIR时只能使用shape和类型与输入相同的数据。
2022-07-22 16:25:38 +08:00
参数:
- **graph** (object) - 从MindIR加载的编译图。
- **params_init** (dict) - 需要在图中初始化的参数。key为参数名称类型为字符串value为 Tensor 或 Parameter。如果参数名在图中已经存在则更新其值如果不存在则忽略。默认值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。