TypeError: expected Tensor as element 0 in argument 0, but got numpy.ndarray
问题描述
原因分析:
需要Tensor变量,我却给了numpy变量,所以转化一下就好啦!!
我们使用torch.Tensor()方式进行转化即可
# 转换成Tensordata0 = torch.Tensor(data0)
需要Tensor变量,我却给了numpy变量,所以转化一下就好啦!!
# 转换成Tensordata0 = torch.Tensor(data0)