PyTorch 中 nn.Sequential 模型实例化与前向传播的正确用法

小伟君_7091

小伟君_7091

2026-05-31

979人浏览

原创

在 PyTorch 中,模型类(如继承 nn.Module 的 Net)的 __init__ 方法仅负责初始化网络结构,不能接收输入数据;输入必须在模型实例化后,通过调用实例(即触发 forward)传入。

在 pytorch 中,模型类(如继承 `nn.module` 的 `net`)的 `__init__` 方法仅负责初始化网络结构,**不能接收输入数据**;输入必须在模型实例化后,通过调用实例(即触发 `forward`)传入。

你遇到的错误:

TypeError: Net.__init__() takes 1 positional argument but 2 were given

根本原因在于这行代码:

model = Net(X_train[:5])  # ❌ 错误:把输入数据传给了构造函数!

Net.__init__() 只接受 self 这一个隐式参数(即 def __init__(self):),而 X_train[:5] 被当作第二个位置参数传入,导致 Python 报错。

✅ 正确做法是先实例化模型,再传入数据进行前向传播:

PyTorch Linux版 2.11.0
PyTorch Linux版 2.11.0

PyTorch 2.11.0 历史版本下载,来自 PyPI 官方发布,适合旧项目兼容、实验复现和指定环境安装。

下载
import torch
import torch.nn as nn

# 定义模型
class Net(nn.Module):
    def __init__(self):
        super().__init__()
        self.stack = nn.Sequential(
            nn.Linear(3, 8),
            nn.ReLU(),
            nn.Linear(8, 8),
            nn.ReLU(),
            nn.Linear(8, 1),
            nn.Sigmoid()
        )

    def forward(self, x):
        return self.stack(x)

# ✅ 正确:先创建模型实例(不传数据)
model = Net()

# ✅ 正确:将输入张量传给模型实例(自动调用 forward)
x_sample = torch.randn(5, 3)  # 假设 batch_size=5, input_dim=3
output = model(x_sample)       # shape: [5, 1]

print(output.shape)  # torch.Size([5, 1])
print(output.squeeze())  # 可选:展平为一维概率输出

⚠️ 注意事项:

  • X_train[:5] 是你的数据切片,它属于 forward 阶段的输入,绝不可出现在 Net(...) 构造调用中;
  • 确保输入张量 x 的第二维(特征数)与首层 nn.Linear(in_features=3, ...) 匹配;
  • 若 X_train 是 NumPy 数组(如 np.ndarray),需先转换为 torch.Tensor 并指定 dtype=torch.float32:
    x_tensor = torch.from_numpy(X_train[:5]).float()
    output = model(x_tensor)

? 小技巧:PyTorch 模型实例是可调用对象(callable),model(x) 等价于 model.forward(x),这是设计使然,也是你无需显式写 .forward() 的原因。

总结:牢记 “构造(__init__)只建图,调用(model(x))才跑数” —— 这是 PyTorch 面向对象建模的核心范式,也是避免此类 TypeError 的关键。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

pytorch

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1731

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

4424

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1709

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

25897

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3147

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3167

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1183

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

596

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2463

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程