news 2026/8/21 15:53:49

探究nn.Linear和nn.Conv2d的基本用法

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
探究nn.Linear和nn.Conv2d的基本用法

1问题

1.如何在pytorch中使用nn.Linear?

2.如何在pytorch环境中运用nn.Conv2d?

2 方法

1.首先在网上查询了关于nn.Linear的含义。PyTorch的nn.Linear()是用于设置网络中的全连接层的,需要注意在二维图像处理的任务中,全连接层的输入与输出一般都设置为二维张量,形状通常为[batch_size, size],不同于卷积层要求输入输出是四维张量。其中in_features指的是输入的二维张量的大小,即输入的[batch_size, size]中的size。out_features指的是输出的二维张量的大小,即输出的二维张量的形状为[batch_size,output_size],当然,它也代表了该全连接层的神经元个数。
从输入输出的张量的shape角度来理解,相当于一个输入为[batch_size, in_features]的张量变换成了[batch_size, out_features]的输出张量。

  1. Conv2d是用来实现2d卷积操作的,其中:

in_channels 表示输入的channels数

out_channels 表示输出的channels数,

kernel_size 表示卷积核的尺寸,可以是方形卷积核、也可以不是,下边example可以看到

stride 表示 步长,用来控制卷积核移动间隔

padding 表示输入边沿扩边操作

import torch
from torch import nn
conv1 = nn.Conv2d(
in_channels=3, #输入通道数
#每个卷积核的通道数与输入通道数相同
#卷积运算后得到128张特征图
out_channels=128, #输出通道数,表示当前卷积层有128个卷积核
kernel_size=3, #卷积核大小3x3
stride=1, #步长,每次滑动窗口移动的长度,默认为1
)
fc1 = nn.Linear(
in_features=128*26*26, #前一层拉伸后的向量大小
out_features=10,
)
x = torch.rand(size=(32,3,28,28))
x = conv1(x)
x = torch.flatten(x,1) #二维的特征图与全连接进行连接时需要拉伸
x = fc1(x)
#[32,3,28,28] -> [32,32,26,26]
print(x.shape)

输出:

3结语

针对探究nn.Linear问题,其用于设置网络中的全连接层的,需要注意在二维图像处理的任务中,全连接层的输入与输出一般都设置为二维张量,其中参数in_features指的是输入的二维张量的大小,即输入的[batch_size, size]中的size。out_features指的是输出的二维张量的大小。

针对nn.Conv2d问题,其用法是用来实现2d卷积操作的,其中包含了许多用于卷积的参数,in_channels、out_channels和kernel_size是需要手动提供的,而其他的参数是有默认值。

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/8/21 22:17:05

黑极光君与面包君的对话10

面包君:你说一种理论体系不能广义化,不能融入生活,解决实际问题,那这不就是屠龙技术吗?现代又没有龙,要这种技术又有何用呢?极光君黑极光君:是的,你这个比喻非常准确&…

作者头像 李华
网站建设 2026/8/21 22:16:34

基于VUE的MBTI人格测试系统 [VUE]-计算机毕业设计源码+LW文档

摘要:MBTI(Myers-Briggs Type Indicator)作为一种广泛应用的人格测评工具,能够帮助人们更好地了解自己和他人的人格特征。本文介绍了一个基于VUE框架开发的MBTI人格测试系统,阐述了系统的需求分析、技术架构、功能模块…

作者头像 李华
网站建设 2026/8/21 22:15:36

基于VUE的汽车出租管理系统 [VUE]-计算机毕业设计源码+LW文档

摘要:随着汽车出租行业的蓬勃发展,传统的管理方式已难以满足高效、精准的业务需求。本文设计并实现了一个基于VUE框架的汽车出租管理系统,详细阐述了系统的需求分析、技术选型、架构设计、功能模块实现等内容。该系统涵盖了系统用户管理、新闻…

作者头像 李华
网站建设 2026/8/21 22:01:49

光储充一体化方案如何进行精准设计

在能源结构改变的推动下,光储充一体化项目成为新能源转型的重要载体,但传统设计中成本高企、配储盲目、收益模糊等痛点,让投资方举棋难定。鹧鸪云光储仿真设计软件以数字化技术打破瓶颈,为光储充方案设计提供全流程科学支撑&#…

作者头像 李华