深度学习篇---深度学习框架


文章目录


前言

以上就是今天要讲的内容,本文仅仅简单介绍了Pytorch、TensorFlow、Paddlepaddle三个深度学习框架以及其基本操作。


第一部分:框架简介

1. PyTorch

简介

简介: PyTorch是由Facebook的人工智能研究团队开发的一个开源机器学习库,主要用于应用如计算机视觉和自然语言处理等领域 的深度学习。它是一个动态计算图框架,提供了灵活性和易用性。

特点

动态计算图

动态计算图(Dynamic Computation Graph): PyTorch使用动态图,使得调试和实验更加灵活。

易于上手

易于上手: PyTorch的API设计接近Python原生代码,易于理解和编写。

强大的社区支持

强大的社区支持: PyTorch拥有一个活跃的社区,提供了大量的预训练模型和教程

与Python的集成度高

与Python深度集成: PyTorch可以很好地与Python的其他库(如NumPy)集成

核心组件

torch.Tensor: 提供了多维数组(张量)的操作。
torch.nn: 包含了
神经网络的各种层和激活函数

torch.optim: 提供了各种优化算法

torch.utils.data: 用于数据加载和预处理

2. TensorFlow

简介

简介: TensorFlow是由Google Brain团队开发的另一个开源机器学习库,它支持广泛的机器学习任务,并且可以在多种平台上运行

特点

静态计算图

静态计算图(Static Computation Graph): TensorFlow使用静态图,有利于优化和部署。

跨平台

跨平台: TensorFlow可以在不同的设备上运行 ,包括CPU、GPU和TPU

强大的生态系统

强大的生态系统: TensorFlow拥有丰富的工具和库,如TensorBoard、TensorFlow Lite等。

Keras集成

Keras集成: TensorFlow 2.x版本中,Keras成为了其高级API,使得构建和训练模型更加简单。

核心组件

tf.Tensor: 表示计算图中的数据

tf.keras: 提供了高层次的API来构建和训练模型

tf.data: 用于数据输入管道

tf.train: 提供了优化器和其他训练工具

3. PaddlePaddle

简介

简介: PaddlePaddle(简称PD)是由百度开发的一个深度学习平台,旨在让研究人员和开发人员能够轻松地构建各种深度学习模型。

特点

易于使用

易于使用: PaddlePaddle提供了丰富的API,使得模型构建更加简单。

高性能

高性能: PaddlePaddle针对服务器和移动设备进行了优化。

工业级应用

工业级应用: PaddlePaddle在工业界有广泛的应用,特别是在中国。

丰富的预训练模型

丰富的预训练模型: PaddlePaddle提供了大量的预训练模型,方便开发者使用。

核心组件

paddle.Tensor: 提供了张量操作。

paddle.nn: 包含了神经网络的层和激活函数

paddle.optimizer: 提供了各种优化算法。

paddle.io: 用于数据加载和预处理

这三个框架各有优势,开发者可以根据自己的需求、熟悉度和项目特点来选择合适的框架。

第二部分:基本操作

PyTorch 基本操作

安装PyTorch:

bash 复制代码
pip install torch torchvision torchaudio

创建张量(Tensor):

python 复制代码
import torch

创建一个未初始化的5x3矩阵

python 复制代码
x = torch.empty(5, 3)

创建一个随机初始化的5x3矩阵

python 复制代码
x = torch.rand(5, 3)

创建一个全零的5x3矩阵,数据类型为long

python 复制代码
x = torch.zeros(5, 3, dtype=torch.long)

张量操作:

创建张量

python 复制代码
x = torch.tensor([5.5, 3])
y = torch.tensor([2.0, 1.0])

加法运算

python 复制代码
z = x + y

乘法运算

python 复制代码
z = x * y

索引

python 复制代码
print(x[0])

自动微分(Autograd):

requires_grad=True表示需要计算梯度

python 复制代码
x = torch.ones(2, 2, requires_grad=True)
y = x + 2
z = y * y * 3
out = z.mean()

计算梯度

python 复制代码
out.backward()

打印梯度

python 复制代码
print(x.grad)

神经网络(nn.Module):

python 复制代码
import torch.nn as nn
import torch.nn.functional as F

class Net(nn.Module):
    def __init__(self):
        super(Net, self).__init__()
        # 定义网络层
        self.conv1 = nn.Conv2d(1, 6, 3)
        self.conv2 = nn.Conv2d(6, 16, 3)

    def forward(self, x):
        # 前向传播
        x = F.max_pool2d(F.relu(self.conv1(x)), (2, 2))
        x = F.max_pool2d(F.relu(self.conv2(x)), 2)
        return x

net = Net()
print(net)

TensorFlow 基本操作

安装TensorFlow:

bash 复制代码
pip install tensorflow

创建张量(Tensor):

python 复制代码
import tensorflow as tf

创建一个常量张量

python 复制代码
x = tf.constant([[1, 2, 3], [4, 5, 6]])

张量操作:

创建张量

python 复制代码
a = tf.constant([[1, 2], [3, 4]])
b = tf.constant([[5, 6], [7, 8]])

加法运算

python 复制代码
c = a + b

乘法运算

python 复制代码
d = a * b

执行运算

python 复制代码
print(c.numpy())

自动微分(Autograph):

使用tf.GradientTape来跟踪操作

python 复制代码
x = tf.Variable(3.0)
with tf.GradientTape() as tape:
    y = x * x

计算梯度

python 复制代码
dy_dx = tape.gradient(y, x)
print(dy_dx.numpy())

构建模型(tf.keras):

python 复制代码
model = tf.keras.Sequential([
    tf.keras.layers.Dense(10, activation='relu', input_shape=(32,)),
    tf.keras.layers.Dense(1)
])

model.compile(optimizer='adam',
              loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),
              metrics=['accuracy'])

模型摘要

python 复制代码
model.summary()

PaddlePaddle 基本操作

安装PaddlePaddle:

bash 复制代码
pip install paddlepaddle

创建张量(Tensor):

python 复制代码
import paddle

创建一个零张量

python 复制代码
x = paddle.zeros([2, 3], dtype='float32')

张量操作:

创建张量

python 复制代码
a = paddle.to_tensor([[1.0, 2.0], [3.0, 4.0]])
b = paddle.to_tensor([[5.0, 6.0], [7.0, 8.0]])

加法运算

python 复制代码
c = paddle.add(a, b)

总结

以上就是今天要讲的内容,本文仅仅简单介绍了Pytorch、TensorFlow、Paddlepaddle三个深度学习框架以及其基本操作。

相关推荐
jinxindeep7 分钟前
CVPR26最佳论文提名:NitroGen,面向通用游戏智能体的 视觉-动作基础模型
人工智能·游戏
小雨下雨的雨4 小时前
井字棋AI机器人实现详解 - Minimax算法实战-鸿蒙PC Electron框架完成
前端·人工智能·算法·华为·electron·鸿蒙
我没胡说八道6 小时前
高校论文AI检测优化工具对比研究与实测分析(2026)
人工智能·深度学习·机器学习·计算机视觉·aigc·论文
秦亚伟6 小时前
AI浪潮重塑融资租赁行业新格局
人工智能
love530love6 小时前
LiveTalking 数字人项目 Windows 部署完全指南(EPGF 架构)
人工智能·windows·python·架构·livetalking·epgf
元启数宇6 小时前
喷淋AI布点实战:8小时人工布点→20分钟自动出图
人工智能
哈哈,柳暗花明6 小时前
人工智能专业术语详解(H)
人工智能·专业术语
圣殿骑士-Khtangc6 小时前
AI 编程工具 2026 实战横评:Cursor 3 vs Claude Code vs Copilot,开发者选型完全指南
人工智能·copilot
云器科技6 小时前
云器Lakehouse 2026年5月版本发布:拥抱 AI Agent,重塑数据智能开发新范式
人工智能
小鹰-上海鹰谷-电子实验记录本6 小时前
第六届党建引领科创生态座谈会 | 邓光辉博士出席分享AI赋能创新药科研新范式
人工智能·ai·电子实验记录本·药企合规