PyTorch——从入门到精通：PyTorch基础知识（张量）【PyTorch系统学习】

什么是张量（Tensor）

张量在数学中是一个代数对象，描述了与矢量空间相关的代数对象集之间的多重线性映射。张量是向量和矩阵概念的推广，可以理解为多维数组。作为数学中的一个基本概念，张量有着多种类型，包括但不限于标量、矢量、矩阵以及矢量空间之间的多重线性映射等。张量的不同类型对应于不同的阶数，其中标量是0阶张量，矢量是1阶张量，矩阵是2阶张量，而更高阶的张量则可以表示更复杂的关系和结构。

|----------|--------------------------------|
| 张量维度 | 代表含义 |
| 0维张量 | 代表的是标量（数字） |
| 1维张量 | 代表的是向量 |
| 2维张量 | 代表的是矩阵 |
| 3维张量 | 时间序列数据股价文本数据单张彩色图片(RGB) |

张量在物理学、工程学、计算机科学等多个领域中都有广泛的应用，特别是在机器学习和深度学习中，张量是一个核心概念，被广泛用于表示和操作图像、文本和时间序列等复杂数据结构。深度学习框架如 TensorFlow 和 PyTorch 就是以张量为核心数据结构，提供了丰富的张量操作函数，使得模型训练和推理变得更加高效和便捷。

PyTorch张量基础操作

由于我们的代码是建立在PyTorch的基础上的，因此首先需要导入torch包，这也是我们后续代码的基础。

复制代码

import torch

创建张量

1.标量：简单的一个数字，只有大小，没有方向的量.

复制代码

# 根据现有数字创建标量，并进行简单的运算操作
x = torch.tensor(3.0)
y = torch.tensor(2.0)

x + y, x - y, x * y, x / y, x**y

2.向量：既有大小，又有方向的量。

复制代码

# 从列表创建张量
tensor_from_list = torch.tensor([1, 2, 3, 4])

# 使用arange生成一个从起始值到结束值（不包括结束值）的等差数列张量，1为其步长
tensor_arange = torch.arange(0, 10, 1)

# 使用linspace创建张量，同样需要开始值和结束值参数，以及生成的数值个数
tensor_linspace = torch.linspace(0, 100, steps=5)

3.矩阵：由行和列组成的数组。

复制代码

# 特定数字的张量，全零张量
zeros_tensor = torch.zeros((2, 3))  # 2行3列

# 全一张量
ones_tensor = torch.ones((3, 3))  # 3行3列

# 随机数字张量
random_tensor = torch.rand((4, 3))  # 4行3列随机数

# 单位矩阵
identity_tensor = torch.eye(3)  # 3x3单位矩阵

# 创建与a相同形状的全零张量
a = torch.tensor([[1, 2], [3, 4]])
zeros_like_a = torch.zeros_like(a)

4.高阶张量

复制代码

# 直接使用数据创建
tensor = torch.tensor([
    [[1, 2, 3], [4, 5, 6], [7, 8, 9]],
    [[10, 11, 12], [13, 14, 15], [16, 17, 18]],
    [[19, 20, 21], [22, 23, 24], [25, 26, 27]]
])  # 创建一个3x3x3的高阶张量

# 使用arange、linespace、zeros创建，如：
tensor_arange=tensor.arange((1,2,3,3)) # 1x2x3x3的高阶张量

张量运算

元素级运算

加、减、乘、除、幂运算

复制代码

a = torch.tensor([[1, 2], [3, 4]])
b = torch.tensor([[5, 6], [7, 8]])

a + b, a - b, a * b, a / b, a ** b, torch.pow(a, 2)  # 加、减、乘、除、指数、每个元素平方

广播机制

广播机制可以自动扩展小张量与大张量进行运算，这种机制的工作方式如下：1.通过适当复制元素来扩展一个或两个数组，以便在转换之后，两个张量具有相同的形状；2.对生成的数组执行按元素操作。

复制代码

a = torch.tensor([[1], [2], [3]]) # 形状(1,3)
b = torch.tensor([10, 20])  # 形状(2,2)

a + b  # 相加后的形状为(2,3)

线性代数运算

复制代码

# 矩阵乘法
a = torch.tensor([[1, 2], [3, 4]])
b = torch.tensor([[5, 6], [7, 8]])
matmul_result = torch.matmul(a, b)
print("矩阵乘法结果：", matmul_result)

# 转置
transpose_result = a.T
print("转置结果：", transpose_result)

# 逆矩阵
square_tensor = torch.tensor([[4.0, 7.0], [2.0, 6.0]])
inv_tensor = torch.inverse(square_tensor)
print("逆矩阵结果：", inv_tensor)

形状操作

复制代码

# 对现有的张量重塑形状，这种方式也能够用于创建高阶张量
reshaped_tensor = tensor.arange(0,10,1)reshape((2, 5))  # 将向量调整为2x5的矩阵

# 将向量展开为一维
a=tensor.zeros((2,2,3))
flattened_tensor = a.flatten()  # 展开为一维

# 张量的拼接，这种方式也能够用于创建高阶张量
a = torch.tensor([[1, 2], [3, 4]])
b = torch.tensor([[5, 6]])
concatenated_tensor = torch.cat([a, b], dim=0)  # 沿第0维拼接
print("拼接结果：", concatenated_tensor)

# 拆分张量
split_tensors = torch.chunk(a, 2, dim=0)  # 沿第0维拆分为2部分
print("拆分结果：", split_tensors)

# 增加维度
a = torch.tensor([1, 2, 3])
expanded_tensor = a.unsqueeze(0)  # 在第0维增加维度
print("增加维度结果：", expanded_tensor)

# 减少维度
a = torch.tensor([[[1, 2], [3, 4]]])
squeezed_tensor = a.squeeze()  # 删除维度为1的轴
print("减少维度结果：", squeezed_tensor)

张量的统计运算

复制代码

# 求和、求积
a = torch.tensor([[1, 2], [3, 4]])
sum_result = torch.sum(a)  # 所有元素求和
prod_result = torch.prod(a)  # 所有元素求积
print("求和结果：", sum_result)
print("求积结果：", prod_result)

# 最大值、最小值
max_result = torch.max(a)  # 最大值
min_result = torch.min(a)  # 最小值
print("最大值：", max_result)
print("最小值：", min_result)

# 按维度聚合
sum_along_dim = torch.sum(a, dim=0)  # 按列求和
print("按维度求和结果：", sum_along_dim)

感谢阅读，希望对你有所帮助~