目前在mindspore适配阶段,算子为NondeterministicInts,
这边是我的C++侧的infershape和infertype的实现部分,我输入只有一个,叫shape,规定了输出的shape,编写st,给定输入的shape为一个Tensor([2,2)]后,
st运行时报错
说需要一个tensor,但是我输入的x已经是tensor了呀,请问是哪儿的问题
解答:
可以这样尝试一下
class Net(nn.Cell):
def __init__(self, x):
super(Net, self).__init__()
self.op = P.NonDeterministicInts()
self.x = x
def construct(self):
return self.op(self.x)
把x作为一个常量tensor, 在__init__里进行赋值