0
点赞
收藏
分享

微信扫一扫

pytorch如何转换tensor的类型dtype


比如现在要将 float 类型的 tensor 转换为 long 类型:

# 两种方法都可以
a = a.long()
a = torch.tensor(a, dtype=torch.long)

同理,如果是改为其他类型,则把 long 替换为对应的类型就行

举报

相关推荐

0 条评论