类型转换:
一般只要在tensor后加long(),int(),double(),float(),byte()等函数就能将tensor进行类型转换。
例如:Torch.LongTensor转换为Torch.FloatTensor,直接使用data.float()即可。
>>> b1 = torch.rand(3,3)
>>> b1
tensor([[1.7498, 0.2052, 0.9352],
[0.1171, 0.2046, 0.1682],
[0.3003, 0.7483, 0.0089]])
>>> b1.long()
tensor([[1, 0, 0],
[0, 0, 0],
[0, 0, 0]])
torch.long():向下取整
查看数据类型:
import torch
x=torch.Tensor([1,2])
print(x)
print(type(x))
print(x.dtype) # x的具体类型
运行结果
tensor([1., 2.])
<class 'torch.Tensor'>
torch.float32