PyTorch 介绍

译者:YAOKE7

校对者:FontTian

Torch张量库介绍

深度学习的所有计算都是在张量上进行的,其中张量是一个可以被超过二维索引的矩阵的一般表示形式。稍后我们将详细讨论这意味着什么。首先,我们先来看一下我们可以用张量来干什么。

  1. # 作者: Robert Guthrie
  2. import torch
  3. import torch.autograd as autograd
  4. import torch.nn as nn
  5. import torch.nn.functional as F
  6. import torch.optim as optim
  7. torch.manual_seed(1)

创建张量

张量可以在Python list形式下通过torch.Tensor()函数创建。

  1. # 利用给定数据创建一个torch.Tensor对象.这是一个一维向量
  2. V_data = [1., 2., 3.]
  3. V = torch.Tensor(V_data)
  4. print(V)
  5. # 创建一个矩阵
  6. M_data = [[1., 2., 3.], [4., 5., 6]]
  7. M = torch.Tensor(M_data)
  8. print(M)
  9. # 创建2x2x2形式的三维张量.
  10. T_data = [[[1., 2.], [3., 4.]],
  11. [[5., 6.], [7., 8.]]]
  12. T = torch.Tensor(T_data)
  13. print(T)

输出

  1. tensor([1., 2., 3.])
  2. tensor([[1., 2., 3.],
  3. [4., 5., 6.]])
  4. tensor([[[1., 2.],
  5. [3., 4.]],
  6. [[5., 6.],
  7. [7., 8.]]])

什么是三维张量?让我们这样想象。如果你有一个向量,那么对这个向量索引就会得到一个标量。如果你有一个矩阵,对这个矩阵索引那么就会得到一个向量。如果你有一个三维张量,那么对其索引就会得到一个矩阵!

针对术语的说明:当我在本教程内使用“tensor”,它针对的是所有torch.Tensor对象。矩阵和向量是特殊的torch.Tensors,他们的维度分别是1和2。当我说到三维张量,我会简洁的使用“3D tensor”。

  1. # 索引V得到一个标量(0维张量)
  2. print(V[0])
  3. # 从向量V中获取一个数字
  4. print(V[0].item())
  5. # 索引M得到一个向量
  6. print(M[0])
  7. # 索引T得到一个矩阵
  8. print(T[0])

输出:

  1. tensor(1.)
  2. 1.0
  3. tensor([1., 2., 3.])
  4. tensor([[1., 2.],
  5. [3., 4.]])

你也可以创建其他数据类型的tensors。默认的数据类型为浮点型。可以使用torch.LongTensor()来创建一个整数类型的张量。你可以在文件中寻找更多的数据类型,但是浮点型和长整形是最常用的。

你可以使用torch.randn()创建一个张量。这个张量拥有随机数据和需要指定的维度。

  1. x = torch.randn((3, 4, 5))
  2. print(x)

输出:

  1. tensor([[[-1.5256, -0.7502, -0.6540, -1.6095, -0.1002],
  2. [-0.6092, -0.9798, -1.6091, -0.7121, 0.3037],
  3. [-0.7773, -0.2515, -0.2223, 1.6871, 0.2284],
  4. [ 0.4676, -0.6970, -1.1608, 0.6995, 0.1991]],
  5. [[ 0.8657, 0.2444, -0.6629, 0.8073, 1.1017],
  6. [-0.1759, -2.2456, -1.4465, 0.0612, -0.6177],
  7. [-0.7981, -0.1316, 1.8793, -0.0721, 0.1578],
  8. [-0.7735, 0.1991, 0.0457, 0.1530, -0.4757]],
  9. [[-0.1110, 0.2927, -0.1578, -0.0288, 0.4533],
  10. [ 1.1422, 0.2486, -1.7754, -0.0255, -1.0233],
  11. [-0.5962, -1.0055, 0.4285, 1.4761, -1.7869],
  12. [ 1.6103, -0.7040, -0.1853, -0.9962, -0.8313]]])

张量操作

你可以以你想要的方式操作张量。

  1. x = torch.Tensor([1., 2., 3.])
  2. y = torch.Tensor([4., 5., 6.])
  3. z = x + y
  4. print(z)

输出:

  1. tensor([5., 7., 9.])

可以查阅 文档 获取大量可用操作的完整列表,这些操作不仅局限于数学操作范围。

接下来一个很有帮助的操作就是连接。

  1. # 默认情况下, 它沿着第一个行进行连接 (连接行)
  2. x_1 = torch.randn(2, 5)
  3. y_1 = torch.randn(3, 5)
  4. z_1 = torch.cat([x_1, y_1])
  5. print(z_1)
  6. # 连接列:
  7. x_2 = torch.randn(2, 3)
  8. y_2 = torch.randn(2, 5)
  9. # 第二个参数指定了沿着哪条轴连接
  10. z_2 = torch.cat([x_2, y_2], 1)
  11. print(z_2)
  12. # 如果你的tensors是不兼容的,torch会报错。取消注释来查看错误。
  13. # torch.cat([x_1, x_2])

输出:

  1. tensor([[-0.8029, 0.2366, 0.2857, 0.6898, -0.6331],
  2. [ 0.8795, -0.6842, 0.4533, 0.2912, -0.8317],
  3. [-0.5525, 0.6355, -0.3968, -0.6571, -1.6428],
  4. [ 0.9803, -0.0421, -0.8206, 0.3133, -1.1352],
  5. [ 0.3773, -0.2824, -2.5667, -1.4303, 0.5009]])
  6. tensor([[ 0.5438, -0.4057, 1.1341, -0.1473, 0.6272, 1.0935, 0.0939, 1.2381],
  7. [-1.1115, 0.3501, -0.7703, -1.3459, 0.5119, -0.6933, -0.1668, -0.9999]])

重构张量

使用.view()去重构张量。这是一个高频方法,因为许多神经网络的神经元对输入格式有明确的要求。你通常需要先将数据重构再输入到神经元中。

  1. x = torch.randn(2, 3, 4)
  2. print(x)
  3. print(x.view(2, 12)) # 重构为2行12列
  4. # 同上。如果维度为-1,那么它的大小可以根据数据推断出来
  5. print(x.view(2, -1))

输出:

  1. tensor([[[ 0.4175, -0.2127, -0.8400, -0.4200],
  2. [-0.6240, -0.9773, 0.8748, 0.9873],
  3. [-0.0594, -2.4919, 0.2423, 0.2883]],
  4. [[-0.1095, 0.3126, 1.5038, 0.5038],
  5. [ 0.6223, -0.4481, -0.2856, 0.3880],
  6. [-1.1435, -0.6512, -0.1032, 0.6937]]])
  7. tensor([[ 0.4175, -0.2127, -0.8400, -0.4200, -0.6240, -0.9773, 0.8748, 0.9873,
  8. -0.0594, -2.4919, 0.2423, 0.2883],
  9. [-0.1095, 0.3126, 1.5038, 0.5038, 0.6223, -0.4481, -0.2856, 0.3880,
  10. -1.1435, -0.6512, -0.1032, 0.6937]])
  11. tensor([[ 0.4175, -0.2127, -0.8400, -0.4200, -0.6240, -0.9773, 0.8748, 0.9873,
  12. -0.0594, -2.4919, 0.2423, 0.2883],
  13. [-0.1095, 0.3126, 1.5038, 0.5038, 0.6223, -0.4481, -0.2856, 0.3880,
  14. -1.1435, -0.6512, -0.1032, 0.6937]])

计算图和自动求导

计算图的思想对于有效率的深度学习编程是很重要的,因为它可以使你不必去自己写反向梯度传播。计算图只是简单地说明了如何将数据组合在一起以输出结果。因为图完全指定了操作所包含的参数,因此它包含了足够的信息去求导。这可能听起来很模糊,所以让我们看看使用Pytorch的基本类:requires_grad。

首先,从程序员的角度来思考。我们在上面刚刚创建的torch.Tensor对象中存储了什么?显然,是数据和结构,也很可能是其他的东西。但是当我们将两个张量相加,我们得到了一个输出张量。这个输出所能体现出的只有数据和结构,并不能体现出是由两个张量加之和得到的(因为它可能是从一个文件中读取的, 也可能是其他操作的结果等)。

如果requires_grad=True,张量对象可以一直跟踪它是如何创建的。让我们在实际中来看。

  1. # 张量对象带有“requires_grad”标记
  2. x =torch.Tensor([1., 2., 3], requires_grad=True)
  3. # 通过requires_grad=True,您也可以做之前所有的操作。
  4. y = torch.Tensor([4., 5., 6], requires_grad=True)
  5. z = x + y
  6. print(z)
  7. # 但是z还有一些额外的东西.
  8. print(z.grad_fn)

输出:

  1. tensor([5., 7., 9.], grad_fn=<AddBackward0>)
  2. <AddBackward0 object at 0x7fd73f1b4c50>

既然变量知道怎么创建的它们。z知道它并非是从文件读取的,也不是乘法或指数或其他运算的结果。如果你继续跟踪 z.grad_fn,你会从中找到x和y的痕迹。

但是它如何帮助我们计算梯度?

  1. # 我们来将z中所有项作和运算
  2. s = z.sum()
  3. print(s)
  4. print(s.grad_fn)

输出:

  1. tensor(21., grad_fn=<SumBackward0>)
  2. <SumBackward0 object at 0x7fd73f1c7e48>

那么这个计算和对x的第一个分量的导数等于多少? 在数学上,我们求

\frac{\partial s}{\partial x_0}s是被作为张量z的和创建的。张量z是x+y的和

s = \overbrace{x_0 + y_0}^\text{z_0} + \overbrace{x_1 + y_1}^\text{z_1} + \overbrace{x_2 + y_2}^\text{z_2}

并且s包含了足够的信息去决定我们需要的导数为1!

当然它掩盖了如何计算导数的挑战。这是因为s携带了足够多的信息所以导数可以被计算。现实中,Pytorch 程序的开发人员用程序指令sum()和 + 操作以知道如何计算它们的梯度并且运行反向传播算法。深入讨论此算法超出了本教程的范围.

让我们用Pytorch计算梯度,发现我们是对的:(注意如果你运行这个模块很多次,它的梯度会上升,这是因为Pytorch累积梯度渐变为.grad属性,而且对于很多模型它是很方便的.)

  1. # 在任意变量上使用 .backward()将会运行反向,从它开始.
  2. s.backward()
  3. print(x.grad)

输出:

  1. tensor([1., 1., 1.])

作为一个成功的深度学习程序员了解下面的模块如何运行是至关重要的。

  1. x = torch.randn((2, 2))
  2. y = torch.randn((2, 2))
  3. #用户创建的张量在默认情况下“requires_grad=False”
  4. print(x.requires_grad, y.requires_grad)
  5. z = x + y
  6. # 你不能通过z反向传播。
  7. print(z.grad_fn)
  8. # “.requires_grad_( ... )”改变了“requires_grad”属性
  9. # 如果没有指定,标记默认为True
  10. x = x.requires_grad_()
  11. y = y.requires_grad_()
  12. #正如我们在上面看到的一样,z包含足够的信息计算梯度。
  13. z = x + y
  14. print(z.grad_fn)
  15. # 如果任何操作的输入部分带有“requires_grad=True”那么输出就会变为:
  16. print(z.requires_grad)
  17. # 现在z有关于x,y的历史信息
  18. # 我们可以获取它的值,将其从历史中分离出来吗?
  19. new_z = z.detach()
  20. # new_z有足够的信息反向传播至x和y吗?
  21. # 答案是没有
  22. print(new_z.grad_fn)
  23. # 怎么会这样? “z.detach()”函数返回了一个与“z”相同存储的张量
  24. #但是没有携带历史的计算信息。
  25. # 它对于自己是如何计算得来的不知道任何事情。
  26. # 从本质上讲,我们已经把这个变量从过去的历史中分离出来了。

输出:

  1. False False
  2. None
  3. <AddBackward0 object at 0x7fd73f1ee588>
  4. True
  5. None

您也可以通过.requires_grad``=True by wrapping the code block in ``with torch.no_grad():停止跟踪张量的历史记录中的自动求导。

  1. print(x.requires_grad)
  2. print((x ** 2).requires_grad)
  3. with torch.no_grad():
  4. print((x ** 2).requires_grad)

输出:

  1. True
  2. True
  3. False