问题原因:
a,b=torch.tensor([1,2,3]),torch.tensor([1,2,3])
lis.append(a)
lis.append(b)
torch.tensor(lis)
不能把一个元素为tensor的list用torch.tensor进行封装
解决
把tensor转为list
a,b=torch.tensor([1,2,3]),torch.tensor([1,2,3])
lis.append(a.tolist())
lis.append(b.tolist())
torch.tensor(lis)