1 Descripción del error
1.1 Entorno del sistema
Entorno de hardware (Ascend/GPU/CPU): Entorno de software de CPU : -- 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 script de entrenamiento consiste en calcular los primeros k valores máximos de la última dimensión de Tensor mediante la construcción de una red de operador único TopK. El guión es el siguiente:
01 class Net(nn.Cell):
02 def __init__(self):
03 super(Net, self).__init__()
04 self.topk = ops.TopK(sorted=False)
05
06 def construct(self, x, k):
07 output = self.topk(x, k)
08 return output
09
10 net = Net()
11 x = Tensor(([[5, 2, 3, 3, 5], [5, 2, 9, 3, 5]]), mindspore.double)
12 k = 5
13 values, indices = net(x, k)
14 print(values, indices)
Copiar1.2.2 Informe de errores
El mensaje de error aquí es el siguiente: