qq_54867072的博客

原创 nn.Unfold函数详解

kernel_size：卷积核的大小，可以是整数或者是元组，如kernel_size=3 表示使用 3x3 的卷积核，kernel_size=(3, 5) 表示使用一个 3x5 的卷积核。其中，L 是展开的局部块数量，等于卷积操作后有效区域的数量，d代表的是维度，通常只有两个维度，分别是图像的高度H和宽度W，spatial_size[d]是是输入张量在维度。假设我们有一个批次大小为1、输入通道为1、图像大小为4x4的图像，并且使用一个3x3的卷积核，步幅为1，没有填充。卷积核在输入张量上滑动的步长。