1 Descripción del error
1.1 Entorno del sistema
Entorno de hardware (Ascend/GPU/CPU): Entorno de software Ascend : -- Versión de MindSpore (fuente o binario): 1.8.0 -- Versión de Python (p. ej., Python 3.7.5): 3.7.6 -- Plataforma y distribución del sistema operativo ( por ejemplo, Linux Ubuntu 16.04): Ubuntu 4.15.0-74-generic -- GCC/Versión del compilador (si se compila desde la fuente):
1.2 Información básica
1.2.1 Guión
El guión de entrenamiento es promediar y reducir el eje 1 mediante la construcción de la red de operadores ReduceMean. El guión es el siguiente:
01 class Net(nn.Cell):
02 def __init__(self, axis, keep_dims):
03 super().__init__()
04 self.reducemean = ops.ReduceMean(keep_dims=keep_dims)
05 self.axis = axis
06 def construct(self, input_x):
07 return self.reducemean(input_x, self.axis)
08 net = Net(axis=(1,), keep_dims=True)
09 x = Tensor(np.random.randn(1, 2, 3, 4, 5, 6, 7, 8, 9), mindspore.float32)
10 out = net(x)
11 print("out shape: ", o