解决TypeError: expected Tensor as element 0 in argument 0, but got numpy.ndarray错误的方法

TypeError: expected Tensor as element 0 in argument 0, but got numpy.ndarray

问题描述

原因分析：

需要Tensor变量，我却给了numpy变量，所以转化一下就好啦！！

我们使用torch.Tensor()方式进行转化即可

# 转换成Tensor
data0 = torch.Tensor(data0)