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 (str) - 指定解密模式,设置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 >>> >>> 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.]]]]