mindspore.load

class 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.nn as nn
>>> from mindspore import Tensor, export, load
>>>
>>> net = nn.Conv2d(1, 1, kernel_size=3, weight_init="ones")
>>> input_tensor = Tensor(np.ones([1, 1, 3, 3]).astype(np.float32))
>>> export(net, input_tensor, file_name="net", file_format="MINDIR")
>>> graph = load("net.mindir")
>>> net = nn.GraphCell(graph)
>>> output = net(input_tensor)
>>> print(output)
[[[[4. 6. 4.]
   [6. 9. 6.]
   [4. 6. 4.]]]]