You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

mindspore.nn.GraphCell.rst 947 B

4 years ago
4 years ago
1234567891011121314151617181920
  1. mindspore.nn.GraphCell
  2. ======================
  3. .. py:class:: mindspore.nn.GraphCell(graph, params_init=None)
  4. 运行从MindIR加载的计算图。
  5. 此功能仍在开发中。目前 `GraphCell` 不支持修改图结构,在导出MindIR时只能使用shape和类型与输入相同的数据。
  6. **参数:**
  7. - **graph** (object) - 从MindIR加载的编译图。
  8. - **params_init** (dict) - 需要在图中初始化的参数。key为参数名称,类型为字符串,value为 Tensor 或 Parameter。如果参数名在图中已经存在,则更新其值;如果不存在,则忽略。默认值:None。
  9. **异常:**
  10. - **TypeError** – 如果图不是FuncGraph类型。
  11. - **TypeError** – 如果 `params_init` 不是字典。
  12. - **TypeError** – 如果 `params_init` 的key不是字符串。
  13. - **TypeError** – 如果 `params_init` 的value既不是 Tensor也不是Parameter。