mindspore.load

mindspore.load(file_name, **kwargs)[源代码]

加载MindIR文件。

返回一个可以由 GraphCell 执行的对象,更多细节参见类 mindspore.nn.GraphCell

参数:
  • file_name (str) - MindIR文件的全路径名。

  • kwargs (dict) - 配置项字典。

    • dec_key (bytes) - 用于解密的字节类型密钥。有效长度为 16、24 或 32。

    • dec_mode (Union[str, function]) - 指定解密模式,设置dec_key时生效。可选项:’AES-GCM’ | ‘AES-CBC’ | 自定义解密函数。默认值:”AES-GCM”。

      • 关于使用自定义解密加载的详情,请查看 教程

返回:

GraphCell,一个可以由 GraphCell 构成的可执行的编译图。

异常:
  • ValueError - MindIR文件名不存在或 file_name 不是string类型。

  • RuntimeError - 解析MindIR文件失败。

样例:

>>> import numpy as np
>>> import mindspore as ms
>>> import mindspore.nn as nn
>>> from mindspore import Tensor
>>> from mindspore import context
>>> context.set_context(mode=context.GRAPH_MODE)
>>>
>>> net = nn.Conv2d(1, 1, kernel_size=3, weight_init="ones")
>>> input_tensor = Tensor(np.ones([1, 1, 3, 3]).astype(np.float32))
>>> ms.export(net, input_tensor, file_name="net", file_format="MINDIR")
>>> graph = ms.load("net.mindir")
>>> net = nn.GraphCell(graph)
>>> output = net(input_tensor)
>>> print(output)
[[[[4. 6. 4.]
   [6. 9. 6.]
   [4. 6. 4.]]]]