python的 list 数据类型转为 pytorch 的tensor数据类型:
如果是多维的list就会报错 :ValueError:only one element tensors can be converted to Python scalars。注意 torch.tensor( XXX ): 语句中的XXX必须是一维的list。
·
注意 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博客
点击阅读全文
更多推荐
所有评论(0)