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

问题描述

原因分析:

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

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

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