pytorch tensor/数据类型转化
Posted 冬日and暖阳
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了pytorch tensor/数据类型转化相关的知识，希望对你有一定的参考价值。
pytorch 怎么进行数据类型的转化呢？
https://pytorch.org/docs/stable/generated/torch.Tensor.to.html
使用类方法 to，可以转化数据类型和数据postion，即可以把tensor在CPU和GPU之前进行转化
tensor = torch.randn(2, 2) # Initially dtype=float32, device=cpu
tensor.to(torch.float64)
cuda0 = torch.device('cuda:0')
tensor.to(cuda0)
tensor.to(cuda0, dtype=torch.float64)
other = torch.randn((), dtype=torch.float64, device=cuda0)
tensor.to(other, non_blocking=True)
以上是关于pytorch tensor/数据类型转化的主要内容，如果未能解决你的问题，请参考以下文章
Pytorch深度学习实战3-2：什么是张量？Tensor的创建与索引
Tensor to img && imge to tensor （pytorch的tensor转换）