forked from huawei/mindspore2022
21 lines
947 B
ReStructuredText
21 lines
947 B
ReStructuredText
mindspore.nn.GraphCell
|
||
======================
|
||
|
||
.. py:class:: mindspore.nn.GraphCell(graph, params_init=None)
|
||
|
||
运行从MindIR加载的计算图。
|
||
|
||
此功能仍在开发中。目前 `GraphCell` 不支持修改图结构，在导出MindIR时只能使用shape和类型与输入相同的数据。
|
||
|
||
**参数：**
|
||
|
||
- **graph** (object) - 从MindIR加载的编译图。
|
||
- **params_init** (dict) - 需要在图中初始化的参数。key为参数名称，类型为字符串，value为 Tensor 或 Parameter。如果参数名在图中已经存在，则更新其值；如果不存在，则忽略。默认值：None。
|
||
|
||
**异常：**
|
||
|
||
- **TypeError** – 如果图不是FuncGraph类型。
|
||
- **TypeError** – 如果 `params_init` 不是字典。
|
||
- **TypeError** – 如果 `params_init` 的key不是字符串。
|
||
- **TypeError** – 如果 `params_init` 的value既不是 Tensor也不是Parameter。
|