这个屌丝很懒，什么也没留下！

pytorch如何转换tensor的类型dtype_tensor改变dtype

作者：凡人多烦事01 | 2024-02-16 16:43:54

tensor改变dtype

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


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

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

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/凡人多烦事01/article/detail/95301