pytorch中一维卷积conv1d,参数含义,为何转置
在Pytorch中,一维卷积是一种非常常用的运算,但是对于刚刚入门的新手,了解此函数的用法仍是一个比较费力的挑战。本文简单介绍一下此函数的常用参数的意义以及此函数的用法。
链接: cov1d函数的官方网站
本文以一个简单的例子作为介绍,现在应用的背景是二维平面上的点,每个batch的size为256,每一个训练数据包含20个点的二维坐标,那么原始输入的shape即为(256,20,2),现在的目的是对二维坐标做一个embedding,使每个点的维度从2维扩展成128维。官网上的部分截图如下所示:

因此这里重点关注 N N N, C C C, L L L的含义即可。 N N N表示batch的size,在本例中,也就是256;C代表通道数,因此本例中 C i n = 2 C_{in}=2

本文介绍了Pytorch中一维卷积conv1d函数的参数意义,包括N(批量大小),C(通道数),L(输入序列长度)。在一个例子中,解释了如何使用conv1d进行二维坐标向量的embedding,将2D扩展到128D。代码示例展示了如何通过转置操作处理输入和输出的维度以适应conv1d函数。

3万+

被折叠的 条评论
为什么被折叠?



