pytorch tensor 数据类型的转换

 tensor  一共包含的数据类型，数据

class DoubleTensor(Tensor): ...
class FloatTensor(Tensor): ...
class LongTensor(Tensor): ...
class IntTensor(Tensor): ...
class ShortTensor(Tensor): ...
class CharTensor(Tensor): ...
class ByteTensor(Tensor): ...
class BoolTensor(Tensor): ...

 例子：

import numpy as np
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.autograd import Variable
b = torch.rand(3,3)#得到的是floattensor值，
print(type(b),b,b.type())
a = b.long()#得到的是longtensor值
print(type(a),a,a.type())
d=b.int()
print(type(d),d,d.type())

运行结果：

PS F:\Graduation project\deeplearning> python .\c.py
tensor([[0.4874, 0.3303, 0.6307],
        [0.4765, 0.2642, 0.4723],
        [0.2643, 0.5587, 0.0493]]) torch.FloatTensor
tensor([[0, 0, 0],
        [0, 0, 0],
        [0, 0, 0]]) torch.LongTensor
tensor([[0, 0, 0],
        [0, 0, 0],
        [0, 0, 0]], dtype=torch.int32) torch.IntTensor

 

总结：

查看数据类型使用a.type() ， 数据转换.long()  其余类似

参考：https://blog.csdn.net/hustchenze/article/details/79154139