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' | 'SM4-CBC' | 'AES-CBC' | 自定义解密函数。默认值:
"AES-GCM"
。关于使用自定义解密加载的详情,请查看 教程。
obf_func (function) - 导入混淆模型所需要的函数,可以参考 obfuscate_model() 了解详情。
- 返回:
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.]]]]
- 教程样例: