# 损失函数
# nn.MSELoss 来计算均方误差
# nn.CrossEntropyLoss 来计算交叉熵损失
output = net(input)
target = Variable(t.arange(0, 10))
# target = target.float()
criterion = nn.MSELoss()
loss = criterion(output, target)
loss
在上面注释代码里加上一行即可
target = target.float()
得到答案
tensor(28.3906, grad_fn=)