1. 程式人生 > 其它 >ValueError: only one element tensors can be converted to Python scalars

ValueError: only one element tensors can be converted to Python scalars

問題原因:

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)