torch GPU数据类型之间的转换
时间: 2023-04-10 12:00:52 浏览: 129
数据类型之间的转换
可以使用torch.Tensor的to()方法进行GPU数据类型之间的转换,例如将CPU上的float类型转换为GPU上的float类型可以使用以下代码:
x = torch.Tensor([1, 2, 3])
x_gpu = x.to(torch.device("cuda:0"))
print(x_gpu.dtype)
输出结果为:torch.float32
阅读全文