A @constexpr python decorator is provided in mindspore.ops.constexpr
, which can be used to decorate a function. The
function will be executed by the Python interpreter in the compiling stage, and finally will be collapsed by constants in the type derivation phase of MindSpore to become a ValueNode of the ANF graph.
Since the function is executed in the compiling state of MindSpore, when using @constexpr to decorate a function, the parameter of this input function must be a constant value that can be determined in compiling state of MindSpore.
Otherwise, if the parameter is a value that cannot be determined in the compiling state, the parameter will be None, which may cause the function output to be different than expected.
When the input parameter can be determined, you can use @constexpr to implement some operations that are not supported in the construct function, for example, operations such as creating tensor based on a certain shape.
To avoid that @constexpr input is a value that cannot be determined in the compiling state, you can perform internal judgment processing on None to avoid some unknown errors.
A code example is as follows:
import numpy as np
from mindspore.ops import constexpr
import mindspore.ops as ops
import mindspore.nn as nn
from mindspore import Tensor
import mindspore
@constexpr
def construct_tensor(x):
if x is None:
raise ValueError("input is an unknown value")
return Tensor(np.array(x), dtype=mindspore.float32)
class Net(nn.Cell):
def __init__(self):
super(Net, self).__init__()
self.relu = ops.ReLU()
def construct(self, x):
return self.relu(construct_tensor(ops.shape(x)))
net = Net()
x = Tensor(np.random.random([7, 6, 3]))
out = net(x)
print(out)
[7. 6. 3.]
As shown below, if we change Net to a value that cannot be determined in the compiling state, an exception will be thrown. Because the input of construct_tensor is a value that can be determined when ReLU is run. ValueError will be thrown in constexpr.
@constexpr
def construct_tensor(x):
if x is None:
raise ValueError("input is an unknown value")
return Tensor(np.array(x), dtype=mindspore.float32)
class Net(nn.Cell):
def __init__(self):
super(Net, self).__init__()
self.relu = ops.ReLU()
def construct(self, x):
return self.relu(construct_tensor(self.relu(x)))
net = Net()
x = Tensor(np.random.random([7, 6, 3]))
out = net(x)
print(out)
The following information is displayed:
ValueError: input is an unknown value
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。