异常代码:
data = [torch.from_numpy(np.array(data_[i][0], data_[i][1])) for i in range(len(data_))]
data_[i]是tensor类型,出错:Cannot convert list to array: ValueError: only one element tensors can be converted to Python scalar
修改:
将data_[i]是tensor类型转为numpy
data = [torch.from_numpy(np.array(data_[i][0].numpy(), data_[i][1].numpy())) for i in range(len(data_))]