报错AttributeError: ‘torch.dtype‘ object has no attribute ‘type‘

AttributeError: ‘torch.dtype’ object has no attribute ‘type’

np.mean()改为torch.mean(torch.stack(my_list))即可

loss是一个tensor
np.mean()输入list可以，array也可以
torch.mean()必须输入tensor,最后得到的结果也是tensor
torch.stack()沿着某个维度进行堆叠数据
所以np.mean()会报错