nn.Conv1d(in_channels=c_in, out_channels=d_model, kernel_size=3)的维度转换理解

本人在项目中经常碰到利用nn.Conv1d实现的全连接层进行维度转化,下面以

nn.Conv1d(in_channels=7, out_channels=512, kernel_size=3, padding=1, padding_mode='circular')
# 96*7——>96*512

为例子介绍维度转化的过程

通常我们首先会将维度转置,然后开始卷积操作
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
最后再转置一下,即可得到结果,实现维度由7到512的转变
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/xxt228/article/details/131088816
今日推荐