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()会报错
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()会报错