极客教程PyTorch 张量(Tensor)，张量是PyTorch最基本的操作对象，英文名称为Tensor，它表示的是一个多维的矩阵。比如零维是一个点，一维就是向量，二维就是一般的矩阵，多维就相当于一个多维的数组，这和numpy是对应的，而且 Pytorch 的 Tensor 可以和 numpy 的ndarray相互转换，唯一不同的是Pytorch可以在GPU上运行，而numpy的 ndarray 只能在CPU上运行。
常用的不同数据类型的 Tensor 如下:
- 32位浮点型
torch.FloatTensor - 64位浮点型
torch.DoubleTensor - 16位整型
torch.ShortTensor - 32位整型
torch.IntTensor - 64位整型
torch.LongTensor
创建tensor矩阵
我们可以通过下面这样的方式来定义一个三行两列给定元素的矩阵，并且显示出矩阵的元素和大小：
import torch
a = torch.Tensor([[2,3], [4,8], [7, 9]])
print('a is: {}'.format(a))
print('a size is {}'.format(a.size()))
执行结果如下：
需要注意的是
torch.Tensor默认的是torch.FloatTensor数据类型
也可以定义我们想要的数据类型，就像下面这样：
import torch
b = torch.LongTensor([[2, 3],[4, 8],[7, 9]])
print('b is: {}'.format(b))
print('b size is {}'.format(b.size()))
执行结果如下：
当然也可以创建一个全是 0 的空Tensor或者取一个正太分布作为随机初始值：
import torch
c = torch.zeros((3, 2))
print('zero tensor: {}'.format(c))
d = torch.randn((3,2))
print('noraml randon is : {}'.format(d))
执行结果如下：
我们也可以像numpy一样通过索引的方式取得其中的元素NumPy 切片和索引，同时也可以改变它的值，比如将a的第一行第二列改变为100。
import torch
a = torch.Tensor([[2, 3],[4, 8],[7, 9]])
a[0, 1] = 100
print('changed a is: {}'.format(a))
执行结果如下：
tensor矩阵与ndarray相互转换
除此之外，还可以在Tensor与numpy.ndarray之间相互转换：
import torch
import numpy as np
b = torch.LongTensor([[2, 3],[4, 8],[7, 9]])
numpy_b = b.numpy()
print('cover to many is \n {}'.format(numpy_b))
e = np.array([[2, 3],[4, 5]])
torch_e = torch.from_numpy(e)
print('from numpy tp torch.Tensor is {}'.format(torch_e))
f_torch_e = torch_e.float()
print('change data type to float tensor: {}'.format(f_torch_e))
执行结果如下：
通过简单的b.numpy()，就能将b转换为numpy数据类型，同时使用torch.from_numpy()就能将 numpy 转换为 tensor，如果需要更改 tensor 的数据类型，只需要在转换后的 tensor 后面加上你需要的类型，比如想将 a 的类型转换为 float，只需a.float()就可以了。
GPU 加速
如果你的电脑支持GPU加速，还可以将 Tensor 放到GPU上。首先通过torch.cuda.is_available()判断一下是否支持GPU，如果想把tensor a放到GPU上，只需a.cuda()就能将tensor a放到GPU上了。
if torch.cuda.is_available():
a_cuda = a.cuda()
print(a_cuda)
极客教程相关文章推荐，欢迎阅读！
PyTorch 环境搭建
Cuda和Cudnn 安装教程
Anaconda 安装教程
PyTorch 是什么