• 1. 数据类型:

在torch中CPU和GPU张量分别有8种数据类型, GPU tensor比CPU tensor多了.cuda

在torch中默认的数据类型是32位浮点型(torch.FloatTensor), 

在程序中使用torch.tensor()生成一个张量, 然后使用.dtype获取张量的数据类型

实例:

import torch 

a = torch.tensor([1.3,4.6])
print('a.dtype:',a.dtype)

• 2. 数据类型之间的相互转化:

方法1:

使用a.int(), a.long(), a.short(), a.float(), a.double(), a.half(), a.char(), a.byte() 进行转化

import torch 

a = torch.tensor([1.3,4.6])
b = a.byte()
print('b:',b)

方法2:

使用torch.type()函数, 直接输入需要转化成的类型

import torch 

a = torch.tensor([1.3,4.6])
b = a.type(torch.short)
print('b:',b)

方法3:

使用type_as() 函数,  该函数的作用是将该tensor的类型转化成另一个tensor的类型

import torch 

a = torch.tensor([1.3,4.6])
b = torch.tensor([2,4],dtype=torch.short)
c = a.type_as(b)
print('c:',c)