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 脚本
训练脚本是通过构建ScatterNdUpdate的单算子网络,使用索引和输入张量的值更新张量值。脚本如下:
01 class Net(nn.Cell):
02 def __init__(self, x):
03 super(Net, self).__init__()
04 self.x = Parameter(x, name="x")
05 self.scatter_nd_update = ops.ScatterNdUpdate()
06
07 def construct(self, indices, updates):
08 output = self.scatter_nd_update(self.x, indices, updates)
09 return output
10 input_x =