pytorch将Tensor转为list
import torchvalue = torch.Tensor([1, 2, 3])print("list:", value.tolist())得到：list: [1.0, 2.0, 3.0]
·
import torch
value = torch.Tensor([1, 2, 3])
print("list:", value.tolist())
得到：list: [1.0, 2.0, 3.0]
为开发者提供学习成长、分享交流、生态实践、资源工具等服务，帮助开发者快速成长。
更多推荐
14
0- 0
已为社区贡献66条内容
所有评论(0)