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])
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])