mindspore.load

mindspore.load(file_name, **kwargs)[source]

Load MindIR.

The returned object can be executed by a GraphCell, see class mindspore.nn.GraphCell for more details.

Parameters

file_name (str) – MindIR file name.
kwargs (dict) –
Configuration options dictionary.
- dec_key (bytes): Byte-type key used for decryption. The valid length is 16, 24, or 32.
- dec_mode (Union[str, function]): Specifies the decryption mode, to take effect when dec_key is set.
  - Option: ‘AES-GCM’, ‘AES-CBC’, ‘SM4-CBC’ or customized decryption. Default: ‘AES-GCM’.
  - For details of using the customized decryption, please check the tutorial.
- obf_func (function): A python function used for loading obfuscated MindIR model, which can refer to obfuscate_model() <https://www.mindspore.cn/docs/en/r2.0.0-alpha/api_python/mindspore/mindspore.obfuscate_model.html> .

Returns

GraphCell, a compiled graph that can executed by GraphCell.

Raises

ValueError – MindIR file does not exist or file_name is not a string.
RuntimeError – Failed to parse MindIR file.

Examples

>>> 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.]]]]