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
猜你喜欢
转载自blog.csdn.net/weixin_37707670/article/details/119731721
今日推荐
周排行