当前位置：首页 > news >正文

PyTorch入门学习（六）：神经网络的基本骨架使用

news 来源：原创 2025/5/7 17:12:46

一、引言

二、创建神经网络骨架

三、执行前向传播

一、引言

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

二、创建神经网络骨架

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

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。

三、执行前向传播

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

x = torch.tensor(1.0)
output = tudui(x)
print(output)

创建一个名为x的张量，其值为1.0，作为输入数据。
通过调用Tudui模型的实例tudui并传递输入数据x，执行前向传播操作。
最后，打印前向传播的输出结果。

完整代码如下：

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深度学习快速入门教程（绝对通俗易懂！）【小土堆】