小编给大家分享一下PyTorch基本数据类型有哪些，相信大部分人都还不怎么了解，因此分享这篇文章给大家参考一下，希望大家阅读完这篇文章后大有收获，下面让我们一起去了解一下吧！
1）Tensor(张量)
Pytorch里面处理的最基本的操作对象就是Tensor（张量），它表示的其实就是一个多维矩阵，并有矩阵相关的运算操作。在使用上和numpy是对应的，它和numpy唯一的不同就是，pytorch可以在GPU上运行，而numpy不可以。所以，我们也可以使用Tensor来代替numpy的使用。当然，二者也可以相互转换。
Tensor的基本数据类型有五种：
32位浮点型：torch.FloatTensor。pyorch.Tensor()默认的就是这种类型。
64位整型：torch.LongTensor。
32位整型：torch.IntTensor。
16位整型：torch.ShortTensor。
64位浮点型：torch.DoubleTensor。
那么如何定义Tensor张量呢？其实定义的方式和numpy一样，直接传入相应的矩阵即可即可。下面就定义了一个三行两列的矩阵：
import torch # 导包 a = torch.Tensor([[1, 2], [3, 4], [5, 6]]) print(a)
不过在项目之中，更多的做法是以特殊值或者随机值初始化一个矩阵，就像下面这样：
import torch # 定义一个3行2列的全为0的矩阵 b = torch.zeros((3, 2)) # 定义一个3行2列的随机值矩阵 c = torch.randn((3, 2)) # 定义一个3行2列全为1的矩阵 d = torch.ones((3, 2)) print(b) print(c) print(d)
Tensor和numpy.ndarray之间还可以相互转换，其方式如下：
Numpy转化为Tensor：torch.from_numpy(numpy矩阵)
Tensor转化为numpy：Tensor矩阵.numpy()
范例如下：
import torch import numpy as np # 定义一个3行2列的全为0的矩阵 b = torch.randn((3, 2)) # tensor转化为numpy numpy_b = b.numpy() print(numpy_b) # numpy转化为tensor numpy_e = np.array([[1, 2], [3, 4], [5, 6]]) torch_e = torch.from_numpy(numpy_e) print(numpy_e) print(torch_e)
之前说过，numpy与Tensor大的区别就是在对GPU的支持上。Tensor只需要调用cuda()函数就可以将其转化为能在GPU上运行的类型。
我们可以通过torch.cuda.is_available()函数来判断当前的环境是否支持GPU，如果支持，则返回True。所以，为保险起见，在项目代码中一般采取“先判断，后使用”的策略来保证代码的正常运行，其基本结构如下：
import torch # 定义一个3行2列的全为0的矩阵 tmp = torch.randn((3, 2)) # 如果支持GPU，则定义为GPU类型 if torch.cuda.is_available(): inputs = tmp.cuda() # 否则，定义为一般的Tensor类型 else: inputs = tmp
2）Variable(变量)
Pytorch里面的Variable类型数据功能更加强大，相当于是在Tensor外层套了一个壳子，这个壳子赋予了前向传播，反向传播，自动求导等功能，在计算图的构建中起的很重要的作用。Variable的结构图如下：
其中最重要的两个属性是：data和grad。Data表示该变量保存的实际数据，通过该属性可以访问到它所保存的原始张量类型，而关于该 variable（变量）的梯度会被累计到.grad 上去。
在使用Variable的时候需要从torch.autograd中导入。下面通过一个例子来看一下它自动求导的过程：
import torch from torch.autograd import Variable # 定义三个Variable变量 x = Variable(torch.Tensor([1, 2, 3]), requires_grad=True) w = Variable(torch.Tensor([2, 3, 4]), requires_grad=True) b = Variable(torch.Tensor([3, 4, 5]), requires_grad=True) # 构建计算图，公式为：y = w * x^2 + b y = w * x * x + b # 自动求导，计算梯度 y.backward(torch.Tensor([1, 1, 1])) print(x.grad) print(w.grad) print(b.grad)
上述代码的计算图为y = w * x^2 + b。对x, w, b分别求偏导为：x.grad = 2wx，w.grad=x^2，b.grad=1。代值检验可得计算结果是正确的。
以上是“PyTorch基本数据类型有哪些”这篇文章的所有内容，感谢各位的阅读！相信大家都有了一定的了解，希望分享的内容对大家有所帮助，如果还想学习更多知识，欢迎关注创新互联成都网站设计公司行业资讯频道！
另外有需要云服务器可以了解下创新互联scvps.cn，海内外云服务器15元起步，三天无理由+7*72小时售后在线，公司持有idc许可证，提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案，具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势，专为企业上云打造定制，能够满足用户丰富、多元化的应用场景需求。