注意 torch.tensor( XXX ) : 语句中的XXX必须是一维的list
如果是多维的list就会报错 :ValueError:only one element tensors can be converted to Python scalars
如果想将多维的list内容组成一个tensor ,可以使用stack() 函数:
ref : ValueError:only one element tensors can be converted to Python scalars_warpin的博客-CSDN博客