基本线性代数运算


在实现深度学习模型时,我们难免会对数据进行操作,因此我们需要部分线性代数相关的内容。这节我们将学习线性代数中的基本数学对象和运算,并通过相关代码来表现它们。

标量

我们称仅包含一个数值的为标量,它可以由只有一个元素的张量表示。下面我们简单实例化两个标量,并执行一些熟悉的算术运算。

import torch

x, y = torch.tensor(3.0), torch.tensor(2.0)
x + y, x * y, x / y, x ** y
(tensor(5.), tensor(6.), tensor(1.5000), tensor(9.))

向量

你可以将向量视为标量值组成的列表,下面我们生成一个向量,并通过索引来访问元素。并像python数组一样,用相关函数来看向量的长度和形状。

x = torch.arange(4)
x
tensor([0, 1, 2, 3])

通过索引获取向量中元素,注意索引是从0开始

x[3]
tensor(3)

通过调用python内置len()函数来访问张量的长度

len(x)
4

我们可以用.shape属性访问向量的形状,该属性列出了张量沿 每个轴的长度。由于向量只有一个轴,形状只有一个元素,即向量的长度。

x.shape
torch.Size([4])

矩阵

矩阵将向量从一维推广到二维,在数学表示中,我们使用$A\in R^{mxn}$来表示矩阵A,A由m行和n列的实值标量组成。其中元素$a_{ij}$属于矩阵A第i行第j列,A的形状是(m,n)或m x n。当矩阵具有相同数量的行和列时,我们将A称为方阵。

我们可以通过reshape(m, n)或view(m, n)将一个长度为mn的向量变成形状为m x n的矩阵。

A = torch.arange(20).view(5, 4)       # 也可以用torch.arange(20).reshape(5, 4)
A
tensor([[ 0,  1,  2,  3],
        [ 4,  5,  6,  7],
        [ 8,  9, 10, 11],
        [12, 13, 14, 15],
        [16, 17, 18, 19]])

我们也可以通过一些函数来生成一些特殊向量

# 构造一个未初始化的5x4矩阵
torch.empty(5, 3)

# 构造一个随机初始化的5x4矩阵
torch.rand(5, 4)       # 值在0到1
torch.randn(5, 4)      # 标准正态分布,值在-1到1

# 构造一个全零矩阵
torch.zeros(5, 4)

# 构造一个全一矩阵
torch.ones(5, 4)

# 直接从数据中构造
torch.tensor([[1, 2],
              [2, 3]])

通过.T属性访问矩阵的转置,转置就是交换矩阵中的行和列

A.T
tensor([[ 0,  4,  8, 12, 16],
        [ 1,  5,  9, 13, 17],
        [ 2,  6, 10, 14, 18],
        [ 3,  7, 11, 15, 19]])

索引和变形

Pytorch的索引与Numpy一样,切片也是左闭右开

A[:, 1:]         # 取所有行,取第一列到最后一列
tensor([[ 1,  2,  3],
        [ 5,  6,  7],
        [ 9, 10, 11],
        [13, 14, 15],
        [17, 18, 19]])

变形在Pytorch中用view实现

print(A.shape)
B = A.view(20)
print(B.shape)

# 指定某个维度为-1时,该维度会被自动计算
C = A.view(2, -1)
print(C.shape)
torch.Size([5, 4])
torch.Size([20])
torch.Size([2, 10])

我们可以通过item()函数将tensor中的value取出作为Python的数值,前提是该tensor内部只有一个数值

A[2, 2].item()
10

张量

前面我们介绍了标量、向量和矩阵,其实它们都是张量的一个表示形式。张量为我们提供了描述具有任意数量轴的n维数组的通用方法,它们的索引机制与矩阵类似。

X = torch.arange(24).reshape(2, 3, 4)
X
tensor([[[ 0,  1,  2,  3],
         [ 4,  5,  6,  7],
         [ 8,  9, 10, 11]],

        [[12, 13, 14, 15],
         [16, 17, 18, 19],
         [20, 21, 22, 23]]])

张量算法的基本性质

标量、向量、矩阵和任意数量轴的张量有⼀些实⽤的属性,如任意按元素的一元运算都不会改变张量的形状。同样,给定具有
相同形状的任意两个张量,任何按元素⼆元运算的结果都将是相同形状的张量。如二元运算Hadamard积(两个矩阵按元素乘法)

A = torch.arange(20, dtype=torch.float32).reshape(5, 4)
B = A.clone() # 通过分配新内存,将A的⼀个副本分配给B
A, A * B
(tensor([[ 0.,  1.,  2.,  3.],
         [ 4.,  5.,  6.,  7.],
         [ 8.,  9., 10., 11.],
         [12., 13., 14., 15.],
         [16., 17., 18., 19.]]),
 tensor([[  0.,   1.,   4.,   9.],
         [ 16.,  25.,  36.,  49.],
         [ 64.,  81., 100., 121.],
         [144., 169., 196., 225.],
         [256., 289., 324., 361.]]))

将张量乘以或加上⼀个标量不会改变张量的形状,其中张量的每个元素都将与标量相加或相乘。

a = 2
X = torch.arange(24).reshape(2, 3, 4)
a + X, (a * X).shape
(tensor([[[ 2,  3,  4,  5],
          [ 6,  7,  8,  9],
          [10, 11, 12, 13]],
 
         [[14, 15, 16, 17],
          [18, 19, 20, 21],
          [22, 23, 24, 25]]]),
 torch.Size([2, 3, 4]))

降维和交换维度

降维

我们可以对任意张量进行的一个有用的操作是计算其元素的和,这可以通过调用函数实现

A.shape, A.sum()
(torch.Size([5, 4]), tensor(190.))

默认情况下,求和函数会沿所有的轴降低张量的维度,使它变为一个标量。我们还可以指定张量沿哪一个轴来通过求和降低维度,指定axis=0将通过汇总所有行的元素降维,axis=1将通过汇总所有列的元素降维,下面以axis=1为例

A_sum_axis1 = A.sum(axis=1)
A_sum_axis1, A_sum_axis1.shape
(tensor([ 6., 22., 38., 54., 70.]), torch.Size([5]))

沿着⾏和列对矩阵求和,等价于对矩阵的所有元素进⾏求和。

A.sum(axis=[0, 1])     # 等同于A.sum()
tensor(190.)

⼀个与求和相关的量是平均值(mean或average)。我们通过将总和除以元素总数来计算平均值。在代码中,
我们可以调⽤函数来计算任意形状张量的平均值。

A.mean(), A.sum() / A.numel()
(tensor(9.5000), tensor(9.5000))

同样,计算平均值的函数也可以沿指定轴降低张量的维度。

A.mean(axis=0), A.sum(axis=0) / A.shape[0]
(tensor([ 8.,  9., 10., 11.]), tensor([ 8.,  9., 10., 11.]))

非降维求和

我们可以通过keepdims=True来保持轴数不变,这种求和有时很有用。

sum_A = A.sum(axis=1, keepdims=True)
sum_A
tensor([[ 6.],
        [22.],
        [38.],
        [54.],
        [70.]])

例如,由于sum_A在对每行进行求和后仍保持两个轴,我们可以通过⼴播将A除以sum_A。

A / sum_A
tensor([[0.0000, 0.1667, 0.3333, 0.5000],
        [0.1818, 0.2273, 0.2727, 0.3182],
        [0.2105, 0.2368, 0.2632, 0.2895],
        [0.2222, 0.2407, 0.2593, 0.2778],
        [0.2286, 0.2429, 0.2571, 0.2714]])

交换维度

tensor.transpose(dim0,dim1)交换维度dim0和维度dim1,所以,对于一个张量X(3维及以上的张量),X.transpose(0,2)等价于X.transpose(2,0)

X = torch.arange(24).reshape(2, 3, 4)
Y = X.transpose(0, 2)
X.shape, Y.shape
(torch.Size([2, 3, 4]), torch.Size([4, 3, 2]))

其他运算

点积

我们已经学习了按元素操作、求和及平均值。另⼀个最基本的操作之⼀是点积。给定两个向量$x, y\in R^{d}$,它们的点积$x^{T}y$ (或⟨x, y⟩)是相同位置的按元素乘积的和

x = torch.arange(4, dtype=torch.float32)
y = torch.ones(4, dtype = torch.float32)
x, y, torch.dot(x, y)
(tensor([0., 1., 2., 3.]), tensor([1., 1., 1., 1.]), tensor(6.))

注意,我们可以通过执行按元素乘法,然后进行求和来表示两个向量的点积

torch.sum(x * y)
tensor(6.)

矩阵-向量积

有了点积的基础,我们就可以理解矩阵-向量积,假设矩阵$A\in R^{mxn}$和向量$x\in R^{n}$,则矩阵向量积Ax是一个长度为m的列向量,矩阵向量积是将矩阵的每一行当成一个向量与x做点积得到。

在代码中,我们通过调用torch.mv(A, x)实现,这里请注意A的列维数必须与x的维数(其⻓度)相同

A.shape, x.shape, torch.mv(A, x)
(torch.Size([5, 4]), torch.Size([4]), tensor([ 14.,  38.,  62.,  86., 110.]))

矩阵-矩阵乘法

如果你已经掌握了点积和矩阵-向量积的知识,那么矩阵-矩阵乘法就会很简单,关于矩阵之间的乘法如果不知,可自行百度。

假设我们有两个矩阵$A\in R^{nxk}$和$B\in R^{kxm}$,则矩阵-矩阵乘法AB可以看作是简单地执⾏m次矩阵-向量积,并将结果拼接在⼀起,形成⼀个n x m矩阵。在下⾯的代码中,我们在A和B上执⾏矩阵乘法。这⾥的A是⼀个5⾏4列的矩阵, B是⼀个4⾏3列的矩阵。两者相乘后,我们得到了⼀个5⾏3列的矩阵

B = torch.ones(4, 3)
torch.mm(A, B)
tensor([[ 6.,  6.,  6.],
        [22., 22., 22.],
        [38., 38., 38.],
        [54., 54., 54.],
        [70., 70., 70.]])

范数

概念

以向量范数为例,设V是数域F上的线性空间,且对于V的任一个向量x,对应一个非负实数$\parallel x \parallel$,满足以下条件:

①正定性:$\parallel x \parallel >= 0$,$\parallel x \parallel = 0$当且仅当$x=0$

②齐次性:$\parallel ax \parallel = |a|\parallel x \parallel$

③三角不等式:对任意$x,y\in V$,都有$\parallel x+y \parallel <= \parallel x \parallel + \parallel y \parallel$

则称$\parallel x+y \parallel$为向量x的范数

下面我们介绍p-范数:
$$\parallel x \parallel_{p} = (\overset{n}{\underset{i=1}{\sum}}|x_{i}|^{p})^{1/p}$$
则2-范数就是向量元素平方和的平方根,其实就是欧几里得距离:
$$\parallel x \parallel_{2} = \sqrt{\overset{n}{\underset{i=1}{\sum}}|x_{i}|^{2}}$$
在代码中,我们可以按如下方式计算向量的2-范数

u = torch.tensor([3.0, -4.0])
torch.norm(u)
tensor(5.)

有时你还会碰到1-范数,它表示为向量元素的绝对值之和:
$$\parallel x \parallel_{1} = \overset{n}{\underset{i=1}{\sum}}|x_{i}|$$
我们通过将绝对值函数和按元素求和函数结合起来实现1-范数

torch.abs(u).sum()
tensor(7.)

类似于向量的2-范数,矩阵$X\in R^{mxn}$的F范数也是矩阵元素平方和的平方根:
$$\parallel X \parallel_{F} = \sqrt{\overset{m}{\underset{i=1}{\sum}}\overset{n}{\underset{j=1}{\sum}}x_{ij}^{2}}$$
F范数满足向量范数的所有性质,它就像是矩阵型向量的2-范数。我们可以调用相同的函数norm()计算矩阵的F范数

torch.norm(torch.ones((4, 9)))
tensor(6.)

最后我将列举三种常见的矩阵p-范数

①$\parallel A \parallel_{1} = \underset{j}{max}(\overset{n}{\underset{i=1}{\sum}}|a_{ij}|)$,称为列和范数;

②$\parallel A \parallel_{2} = \sqrt{\lambda_{1}}$,$\lambda_{1}$为$A^{H}A$的最大特征值,称为谱范数

③$\parallel A \parallel_{\infty} = \underset{i}{max}(\overset{n}{\underset{j=1}{\sum}}|a_{ij}|)$,称为行和范数

参考资料

[1] Dive into Deep Learning


文章作者: 不才叶某
版权声明: 本博客所有文章除特別声明外,均采用 CC BY 4.0 许可协议。转载请注明来源 不才叶某 !
评论
  目录