本文重点

前面我们学习了pytorch的默认数据类型是torch.FloatTensor,那么本节课程我们将学习如何将不同的数据类型之间进行数据转换？还会学习tensor中cpu和gpu的转换，以及tensor和numpy以及python之间的类型转换，在实际编程中，可能这些需要时刻使用的。

不同类型之间的转换

不同类型之间的转换非常简单，只需要在Tensor后加 .long(), .int(), .float(), .double()等即可，也可以用.to()函数进行转换

tensor = torch.Tensor(3, 5)

newtensor = tensor.long()——将tensor投射为long类型

newtensor = tensor.half()——将tensor投射为半精度浮点类型

newtensor = tensor.int()——将该tensor投射为int类型

newtensor = tensor.double()——将该tensor投射为double类型

newtensor = tensor.float()——将该tensor投射为float类型

newtensor = tensor.char()——将该tensor投射为char类型

newtensor = tensor.byte()——将该tensor投射为byte类型

newtensor = tensor.short()将该tensor投射为short类型

CPU和GPU之间的转换

CPU张量 ----> GPU张量，使用data.cuda()

GPU张量 ----> CPU张量，使用data.cpu()