注意 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博客

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐