PyTorch基础教程：从张量操作到GPU加速

2025-06-19 09:40:29作者：宗隆裙

PyTorch作为当前最流行的深度学习框架之一，以其动态计算图和直观的Python接口赢得了广大研究者和开发者的青睐。本文将系统性地介绍PyTorch的核心基础知识，帮助初学者快速掌握这一强大工具。

一、PyTorch概述

PyTorch是由Facebook AI Research实验室开发的开源机器学习库，它提供了构建和训练神经网络的灵活接口。与其他框架相比，PyTorch最显著的特点是它的"定义即运行"（define-by-run）机制，这使得模型开发过程更加直观和灵活。

PyTorch的主要优势包括：

动态计算图：允许在运行时修改网络结构
纯Python接口：与Python生态无缝集成
GPU加速支持：充分利用硬件加速
自动微分：简化梯度计算过程
丰富的工具库：支持从研究到生产的全流程

二、张量(Tensor)基础

张量是PyTorch中最基本的数据结构，可以看作是NumPy数组的增强版，支持GPU加速和自动微分功能。

2.1 创建张量

PyTorch提供了多种创建张量的方式：

import torch

# 从Python列表创建
x = torch.tensor([1, 2, 3, 4])

# 创建特定形状的张量
zeros = torch.zeros(3, 4)  # 3行4列的全0张量
ones = torch.ones(2, 3)    # 2行3列的全1张量
rand = torch.rand(2, 2)    # 2×2的均匀分布随机数

# 创建特定范围的张量
range_tensor = torch.arange(0, 10, step=1)  # 0到9的序列

2.2 张量属性

每个PyTorch张量都有一些重要的属性：

x = torch.randn(3, 4, 5)

print(x.shape)      # 张量形状：torch.Size([3, 4, 5])
print(x.dtype)      # 数据类型：torch.float32
print(x.device)     # 存储设备：cpu或cuda

2.3 张量索引与切片

PyTorch支持类似NumPy的索引和切片操作：

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

# 获取单个元素
print(x[0, 0])      # 输出1

# 切片操作
print(x[:, 0])      # 第一列：[1, 4, 7]

# 布尔索引
mask = x > 5
print(x[mask])      # 输出大于5的元素

三、张量运算

PyTorch提供了丰富的张量运算功能，包括算术运算、矩阵运算和归约运算等。

3.1 基本算术运算

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

# 逐元素加法
print(a + b)        # 输出[5, 7, 9]

# 逐元素乘法
print(a * b)        # 输出[4, 10, 18]

3.2 矩阵运算

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

# 矩阵乘法
print(a @ b)        # 使用@运算符
print(torch.matmul(a, b))  # 使用matmul函数

# 矩阵转置
print(a.t())

3.3 归约运算

x = torch.tensor([[1, 2, 3], [4, 5, 6]])

# 求和
print(x.sum())              # 所有元素和
print(x.sum(dim=0))         # 沿行求和
print(x.sum(dim=1))         # 沿列求和

# 最大值
print(x.max(dim=0))         # 返回值和索引

四、与NumPy的互操作

PyTorch与NumPy可以无缝转换，共享内存：

import numpy as np

# NumPy数组转PyTorch张量
np_array = np.array([1, 2, 3])
tensor = torch.from_numpy(np_array)

# PyTorch张量转NumPy数组
tensor = torch.tensor([4, 5, 6])
np_array = tensor.numpy()

注意：CPU上的张量与NumPy数组共享内存，修改一个会影响另一个。

五、GPU加速

PyTorch支持使用GPU加速计算：

# 检查GPU是否可用
print(torch.cuda.is_available())

# 将张量移动到GPU
if torch.cuda.is_available():
    device = torch.device("cuda")
    x = torch.tensor([1, 2, 3], device=device)
    
    # 执行GPU运算
    y = x * 2  # 在GPU上执行

六、计算图与自动微分

PyTorch的动态计算图是其核心特性之一：

# 创建需要计算梯度的张量
x = torch.tensor(2.0, requires_grad=True)
y = torch.tensor(3.0, requires_grad=True)

# 构建计算图
z = x**2 + y**3

# 计算梯度
z.backward()

# 查看梯度
print(x.grad)  # dz/dx = 4
print(y.grad)  # dz/dy = 27