RuntimeError: mean(): input dtype should be either floating point or complex dtypes. Got Long instea

问题描述

在pytorch中创建的张量，求均值的时候报错
RuntimeError: mean(): input dtype should be either floating point or complex dtypes. Got Long instead.

解决办法

将原代码的
torch.tensor
改成
torch.Tensor
结束