pytorch如何转换tensor的类型dtype
比如现在要将 float 类型的 tensor 转换为 long 类型：# 两种方法都可以a = a.long()a = torch.tensor(a, dtype=torch.long)同理，如果是改为其他类型，则把 long 替换为对应的类型就行
·
比如现在要将 float 类型的 tensor 转换为 long 类型：
# 两种方法都可以
a = a.long()
a = torch.tensor(a, dtype=torch.long)同理，如果是改为其他类型，则把 long 替换为对应的类型就行
为开发者提供学习成长、分享交流、生态实践、资源工具等服务，帮助开发者快速成长。
更多推荐
9
0- 0
已为社区贡献1条内容
所有评论(0)