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

ValueError: only one element tensors can be converted to Python scalar
使用torch.tensor 将list列表转化成pytorch中的tensor张量时,如果遇到如下报错,很有可能是因为list中包含多个list数据,或者维度不相同的数据。

使用torch.stack 将list中数据转化成tensor。

torch.tensor([a,b])
#换成stack
torch.stack([a,b])

你可能感兴趣的:(pytorch)