1 报错描述
1.1 系统环境
Hardware Environment(Ascend/GPU/CPU): GPU Software Environment: -- MindSpore version (source or binary): 1.6.0 -- Python version (e.g., Python 3.7.5): 3.7.6 -- OS platform and distribution (e.g., Linux Ubuntu 16.04): Ubuntu 4.15.0-74-generic -- GCC/Compiler version (if compiled from source):
1.2 基本信息
1.2.1 脚本
训练脚本是通过构建StandardNormal的单算子网络,生成符合正态分布的随机数。脚本如下:
01 class Net(nn.Cell):
02 def __init__(self, seed=2, seed2=-3):
03 super(Net, self).__init__()
04 self.standard_normal = ops.StandardNormal(seed=seed, seed2=seed2)
05 def construct(self, output_shape):
06 output = self.standard_normal(output_shape)
07 return output
08
09 output_shape = (2, 3, 4)
10 net = Net()
11 output = net(output_shape)
12 print("OUTPUT: ", output)
复制