【深度学习】S1 预备知识 P1 张量

张量

张量（Tensor）是深度学习中用于表示和处理多维数据的数据结构。张量与 Numpy 类似，但是不同的是深度学习框架专门为张量提供了丰富的操作API，包括创建张量、数据转换、数学运算、索引和切片等，使张量可以使用 GPU 加速，大大提高处理大量数据时的计算速度；同时提供自动微分 Autograd，自动计算和应用梯度。

简单的说，张量（Tensor）就是为深度学习框架量身定制的数据结构。使用张量，深度学习在处理数据和模型计算上更加方便和高效。

---

创建第一个张量

张量的下载与引入

张量是深度学习框架的数据结构，读者首先需要选择一种深度学习框架（TensorFlow、PyTorch…）并下载，本博文基于 PyTorch 深度学习框架。

1. 下载 PyTorch 框架：

# 首先下载PyTorch框架
pip install torch==1.12.0
pip install torchvision==0.13.0
# 本bash执行于Anaconda Prompt
1
2
3
4

2. 引入 Tensor 张量：

# 建议读者使用PyCharm或者VSCode IDE工具
import torch
1
2

创建第一个张量

第一个张量我们使用 arange 创建一个行向量 x：

x = torch.arange(12)
print(x)
1
2

tensor([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11])
1

x 为创建的包含 12 个整数元素的张量。

---

张量的大小和形状

张量的形状

通过张量的 shape 属性来访问张量的形状。

print(x.shape)
1

torch.Size([12])
1

张量的大小

如果只想知道张量中元素的数量，可以使用 numel() 函数。

print(x.numel())
1

12
1

---

改变张量的形状

如果只想改变张量的形状而不改变张量元素数量和值，可以通过使用 reshape() 函数。

y = x.reshape(3,4)
print(y)
1
2

tensor([[ 0,  1,  2,  3],
        [ 4,  5,  6,  7],
        [ 8,  9, 10, 11]])
1
2
3

把张量x从形状（12，）的行向量转换为形状为（3，4）的矩阵y。新的张量形状发生改变，但是元素值并没有改变。

张量的转换也支持自动计算维度的功能，比如：

# 下列 z_1, z_2 张量同上述 reshape(3, 4) 函数功能
z_1 = x.reshape(-1, 4)
z_2 = x.reshape(3, -1)
1
2
3

---

创建特殊张量

有时，我们希望创建全0、全1张量，或者从特定随机分布中随机采样的数字来初始化矩阵，如下。

创建全 0 张量：

# 创建一个形状为 (2, 3, 4) 、值全部为 0 的三维张量
import torch
x_1 = torch.zeros(2,3,4)
print(x_1)
1
2
3
4

tensor([[[0., 0., 0., 0.],
         [0., 0., 0., 0.],
         [0., 0., 0., 0.]],

        [[0., 0., 0., 0.],
         [0., 0., 0., 0.],
         [0., 0., 0., 0.]]])
1
2
3
4
5
6
7

创建全 1 张量：

import torch
x_2 = torch.ones(2,3,4)
print(x_2)
print(x_2.shape)
1
2
3
4

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

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

创建正态分布随机采样张量：
创建一个形状为（3，4）的张量，其中每个元素从均值为 0， 标准差为 1 的标准高斯分布中随机采样。

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

tensor([[ 0.8532, -0.9053, -0.6921,  0.9674],
        [-0.8108, -1.4881, -0.0434,  0.1967],
        [-1.7507, -0.4498,  1.3718, -0.6994]])
1
2
3

---

张量的基本运算

按元素运算

数学运算中最常见的标准算数运算符（+，-，*，/和**）。将两个张量数组作为输入，按元素运算将二元运算符应用于两个数组中的每对位置对应的元素：

import torch

x = torch.tensor([1.0, 2, 4, 8])
y = torch.tensor([2, 2, 2, 2])
print("x+y=", x+y, "\nx-y=", x-y, "\nx*y=", x*y, "\nx/y=", x/y, "\nx**y=", x**y)
1
2
3
4
5

x+y= tensor([ 3.,  4.,  6., 10.]) 
x-y= tensor([-1.,  0.,  2.,  6.]) 
x*y= tensor([ 2.,  4.,  8., 16.]) 
x/y= tensor([0.5000, 1.0000, 2.0000, 4.0000]) 
x**y= tensor([ 1.,  4., 16., 64.])
1
2
3
4
5

按元素计算很重要的一点就是两个张量的形状 shape 必须相同，否则将因找不到对应位置的张量从而无法计算。

张量连结

除了按元素计算外，还可以将多个张量连结在一起 cat(tensors, dim=)，端对端叠起来形成更大的张量。

import torch

x = torch.arange(12).reshape(3, 4)
y = torch.tensor([[2.0, 1, 4, 3], [1, 2, 3, 4], [4, 3, 2, 1]])
print(torch.cat((x,y), dim=1))
1
2
3
4
5

将张量 x（3，4）与张量 y（3，4）横向连结形成更大的张量，形状（3，8）。

tensor([[ 0.,  1.,  2.,  3.,  2.,  1.,  4.,  3.],
        [ 4.,  5.,  6.,  7.,  1.,  2.,  3.,  4.],
        [ 8.,  9., 10., 11.,  4.,  3.,  2.,  1.]])
1
2
3

dim=0 的连结结果将为形状（6，4）。

逻辑运算

通过逻辑运算构建张量，对于两个张量的每个位置，如果对照位置相同，则对应位置值为 True，否则为 False；

import torch

x = torch.arange(12).reshape(3, 4)
y = torch.tensor([[2.0, 1, 4, 3], [1, 2, 3, 4], [4, 3, 2, 1]])

print("x:", x)
print("y:", y)
print("x==y:", x==y)
1
2
3
4
5
6
7
8

x: tensor([[ 0,  1,  2,  3],
        [ 4,  5,  6,  7],
        [ 8,  9, 10, 11]])
y: tensor([[2., 1., 4., 3.],
        [1., 2., 3., 4.],
        [4., 3., 2., 1.]])
x==y: tensor([[False,  True, False,  True],
        [False, False, False, False],
        [False, False, False, False]])
1
2
3
4
5
6
7
8
9

张量求和

如果想对张量中所有元素进行求和，使用 sum() 函数；

print("x:", x)
print(x.sum())
1
2

x: tensor([[ 0,  1,  2,  3],
        [ 4,  5,  6,  7],
        [ 8,  9, 10, 11]])
tensor(66)
1
2
3
4

---

张量的广播机制

在上述张量的基本运算中，我们很多操作都规定需要两个张量保持相同的形状才能完成运算。但是在广播机制中，则不同。张量的广播机制用于扩展张量，虽然不要求两个向量完全相同，但是要求两个向量要分别满足形状：（x, y）与（y, z）方才可以进行运算。

$e.g.$

import torch

a = torch.arange(3).reshape(3, 1)
b = torch.arange(2).reshape(1, 2)

print("a=", a)
print("b=", b)
print("a+b=", a+b)
print("a*b=", a*b)
1
2
3
4
5
6
7
8
9

a= tensor([[0],
        [1],
        [2]])
b= tensor([[0, 1]])
a+b= tensor([[0, 1],
        [1, 2],
        [2, 3]])
a*b= tensor([[0, 0],
        [0, 1],
        [0, 2]])
1
2
3
4
5
6
7
8
9
10

上述张量的广播机制操作中，我们将张量 a（3，1）与张量 b（1，2）广播扩展为一个更大的张量，形状（3，2）。

---

张量的索引与切片

索引、切片访问张量

如 Python 中数组一样，张量中的元素支持通过索引访问。与其他数组一样，张量中第一个元素的索引值为 0，最后一个元素的索引值为 -1。

import torch

x = torch.arange(12)
print("x=", x)
print("x[0]=", x[0], "\nx[-1]=", x[-1])
# 通过切片方式访问张量中的元素值
print("x[1:3]=", x[1:3])
1
2
3
4
5
6
7

x= tensor([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11])
x[0]= tensor(0) 
x[-1]= tensor(11)
x[1:3]= tensor([1, 2])
1
2
3
4

当然张量的操作支持通过索引改变其中元素的值；

import torch

x = torch.arange(12)
print(x)
x[-2] = 22
print(x)
1
2
3
4
5
6

tensor([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11])
tensor([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 22, 11])
1
2

通过切片方式节省内存占用

在张量中，运行一些操作可能会导致需要为新结果分配内存。在深度学习框架张量中，我们可以通过 id() 函数来访问一个张量占用内存的物理位置。

import torch

x = torch.arange(12)
print(id(x))
x = x + 1
print(id(x))
1
2
3
4
5
6

2606440820144
2606440818944
1
2

打印发展张量 x 在计算前后更换了占用的内存位置，或者说，新计算后为张量 x 新分配了内存空间存储其结果。

这种新分配内存空间不仅产生了不必要的内存占用，还会因为其他引用旧的内存位置导致出错。为了避免这种错误，通过切片方式原地更新；

import torch

x = torch.arange(12)
print(id(x))
x[:] = x + 1
print(id(x))
1
2
3
4
5
6

2296319225264
2296319225264
1
2

观察发现，运算前后占用的内存地址空间相同。

---

张量转换为其他对象

Torch 张量与 Numpy 数组共享底层内存，可以就地操作转换；

张量转换为数组：
a.item(), float(a), int(a)：

import torch

a = torch.arange(12)
b = a.numpy()
print(type(a), type(int(a[1])), type(b))
1
2
3
4
5

<class 'torch.Tensor'> <class 'int'> <class 'numpy.ndarray'>
1

数组转换为张量：

import numpy as np
import torch

x = np.arange(12)
y = torch.tensor(x)
print(type(x), type(y))
1
2
3
4
5
6

<class 'numpy.ndarray'> <class 'torch.Tensor'>
1

---

以上内存便是预备知识的第一部分张量的相关介绍；
如有任何问题，请联系或者留言；

2024.2.12