news 2026/10/31 13:57:29

探究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/10/31 8:24:30

终极微博备份指南:Speechless免费工具完整使用教程

终极微博备份指南:Speechless免费工具完整使用教程 【免费下载链接】Speechless 把新浪微博的内容,导出成 PDF 文件进行备份的 Chrome Extension。 项目地址: https://gitcode.com/gh_mirrors/sp/Speechless 在信息碎片化的今天,微博承…

作者头像 李华
网站建设 2026/10/31 18:50:03

暗黑破坏神2存档编辑器终极指南:从零基础到精通进阶

暗黑破坏神2存档编辑器终极指南:从零基础到精通进阶 【免费下载链接】d2s-editor 项目地址: https://gitcode.com/gh_mirrors/d2/d2s-editor 你是否曾经为暗黑破坏神2中的角色Build优化而苦恼?是否想要快速测试不同装备组合的效果却受限于漫长的…

作者头像 李华
网站建设 2026/10/31 1:02:01

LobeChat Google Gemini Pro接入方法:多模态能力整合

LobeChat 与 Google Gemini Pro 的多模态整合实践 在生成式 AI 快速演进的今天,用户对智能助手的期待早已超越“能聊天”的基本功能。我们不再满足于仅用文字提问、等待文本回复——而是希望上传一张产品截图就能获得详细分析,或是拖入一份 PDF 合同便能…

作者头像 李华
网站建设 2026/10/31 6:25:34

LobeChat用量统计面板:跟踪Token消耗与GPU使用率

LobeChat用量统计面板:跟踪Token消耗与GPU使用率 在大模型应用日益普及的今天,一个看似简单的“聊天框”背后,往往隐藏着复杂的资源调度与成本控制挑战。当企业开始将 LLM 集成到客服系统、知识库或自动化流程中时,人们很快意识到…

作者头像 李华
网站建设 2026/10/31 16:20:12

基于VUE的企业咨询管理系统 [VUE]-计算机毕业设计源码+LW文档

摘要:随着企业咨询业务的不断发展和复杂化,传统的管理方式已难以满足高效、精准的业务需求。本文介绍了一种基于VUE框架开发的企业咨询管理系统,详细阐述了系统的需求分析、技术选型、架构设计、功能模块实现等内容。该系统涵盖了系统用户管理…

作者头像 李华