torch 数据类型转换

du = torch.ones([2,2])
a = np.array([[1,2],[3,4]],dtype=np.float32)
b = torch.from_numpy(a)#数据类型是不变的 上面dtype是什么类型 下面torch对应就是什么tensor
print('yes')
#tensor 默认是float32的 比如torch.ones zeros都是生成float32数据a=torch.FloatTensor( (3,2) )
print(a.dtype)
qq = np.array([1,2],dtype=np.int32)
qq = qq.astype(np.float32) #这个是numpy的类型转换 下面是tensor的类型转换 只astype或者int()并不能改变数据类型
#必须有赋值操作才可以a = a.int()##tensor 数据类型转换
print(a.dtype)#对于python的数据 直接进行类型转化即可
#例如
m = 5
n = float(m)

torch 数据类型转换：
tensor = torch.Tensor(3, 5)

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

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

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

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

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

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

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

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