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

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐