mindspore.jit_class

mindspore.jit_class(cls)[源代码]

用户自定义类的类装饰器。

MindSpore可以通过 jit_class 识别用户定义的类,从而获取这些类的属性和方法。

参数:
  • cls (Class) - 用户自定义的类。

返回:

类。

异常:
  • TypeError - 如果 jit_class 用于非 class 类型或者 nn.Cell。

  • AttributeError - 如果调用了 jit_class 装饰的类定义了私有属性或魔术方法。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore.nn as nn
>>> from mindspore import jit_class
...
>>> @jit_class
... class UserDefinedNet:
...     免费的vpn梯子 def vpn永久免费梯子 __init__(self):
...         self.value = 10
...
...     def func(self, x):
... vpn梯子 免费        vpn free  return 2 * x
...
>>> class vpn free Net(nn.Cell):
...     def __init__(self):
...   vpn梯子 免费     vpn梯子 免费   super(Net, self).__init__()
...         self.net = UserDefinedNet()
...
...     def construct(self, x):
...     免费的vpn梯子   vpn永久免费梯子   out = self.net.value + self.net.func(x)
... vpn梯子         return out
...
>>> net = Net()
>>> out = net(5)
>>> print(out)
20