我们经常将数据放到列表里面,
列表中的数据转换为tensor的时候会出现错误:
ValueError: only one element tensors can be converted to Python scalars
原因是,列表中 数据维数过多,
new_data = torch.tensor([item.detach().numpy() for item in data])
data是我们的列表,new_data是tensor啦!
我们经常将数据放到列表里面,
列表中的数据转换为tensor的时候会出现错误:
原因是,列表中 数据维数过多,
new_data = torch.tensor([item.detach().numpy() for item in data])
data是我们的列表,new_data是tensor啦!