PyTorch入门学习(六):神经网络的基本骨架使用

目录

一、引言

二、创建神经网络骨架

三、执行前向传播


一、引言

神经网络是深度学习的基础。在PyTorch中,可以使用nn.Module类创建自定义神经网络模型。本文将演示如何创建一个简单的神经网络骨架并执行前向传播操作。

二、创建神经网络骨架

首先,导入PyTorch库并创建一个继承自nn.Module的自定义神经网络模型:

python 复制代码
import torch
from torch import nn

class Tudui(nn.Module):
    def __init__(self):
        super().__init__()

    def forward(self, input):
        output = input + 1
        return output

tudui = Tudui()
  • 导入PyTorch库以便使用深度学习工具。
  • 创建一个名为Tudui的自定义神经网络模型,它继承自nn.Module
  • __init__方法中,调用父类的构造函数,初始化神经网络。
  • forward方法定义神经网络的前向传播过程,其中对输入进行了简单的操作,将输入加1。

三、执行前向传播

接下来,执行前向传播操作,将输入数据传递给神经网络模型:

python 复制代码
x = torch.tensor(1.0)
output = tudui(x)
print(output)
  • 创建一个名为x的张量,其值为1.0,作为输入数据。
  • 通过调用Tudui模型的实例tudui并传递输入数据x,执行前向传播操作。
  • 最后,打印前向传播的输出结果。

完整代码如下:

python 复制代码
import torch
from torch import nn

# 创建一个自定义神经网络模型 Tudui
class Tudui(nn.Module):
    def __init__(self):
        super().__init__()

    def forward(self, input):
        # 前向传播函数,将输入加1并返回
        output = input + 1
        return output

# 创建 Tudui 类的实例 tudui
tudui = Tudui()

# 创建一个张量 x,值为1.0,作为输入数据
x = torch.tensor(1.0)

# 将输入 x 传递给 tudui 模型,执行前向传播
output = tudui(x)

# 打印前向传播的输出结果
print(output)

参考资料:

视频教程:PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】

相关推荐
云上艺旅5 小时前
K8S学习之基础七十四:部署在线书店bookinfo
学习·云原生·容器·kubernetes
你觉得2056 小时前
哈尔滨工业大学DeepSeek公开课:探索大模型原理、技术与应用从GPT到DeepSeek|附视频与讲义下载方法
大数据·人工智能·python·gpt·学习·机器学习·aigc
hyshhhh6 小时前
【算法岗面试题】深度学习中如何防止过拟合?
网络·人工智能·深度学习·神经网络·算法·计算机视觉
A旧城以西7 小时前
数据结构(JAVA)单向,双向链表
java·开发语言·数据结构·学习·链表·intellij-idea·idea
无所谓จุ๊บ7 小时前
VTK知识学习(50)- 交互与Widget(一)
学习·vtk
FAREWELL000757 小时前
C#核心学习(七)面向对象--封装(6)C#中的拓展方法与运算符重载: 让代码更“聪明”的魔法
学习·c#·面向对象·运算符重载·oop·拓展方法
Listennnn7 小时前
优雅的理解神经网络中的“分段线性单元”,解剖前向和反向传播
人工智能·深度学习·神经网络
吴梓穆7 小时前
UE5学习笔记 FPS游戏制作38 继承标准UI
笔记·学习·ue5
誉鏐7 小时前
PyTorch复现线性模型
人工智能·pytorch·python
Three~stone8 小时前
MySQL学习集--DDL
数据库·sql·学习