参考链接
- https://tangshusen.me/Dive-into-DL-PyTorch/#/chapter05_CNN/5.3_channels
多输入通道
当输入数据含多个通道时,我们需要构造一个输入通道数与输入数据的通道数相同的卷积核,从而能够与含多通道的输入数据做互相关运算。
由于输入和卷积核各有cic_{i}ci个通道,我们可以在各个通道上对输入的二维数组和卷积核的二维核数组做互相关运算,再将这cic_{i}ci个互相关运算的二维输出按通道相加,得到一个二维数组。这就是含多个通道的输入数据与多输入通道的卷积核做二维互相关运算的输出。
def corr2d_multi_in(X, K):
# 沿着X和K的第0维(通道维)分别计算再相加
res = corr2d(X[0, :, :], K[0, :, :])
for i in range(1, X.shape[0]):
res += d2l.corr2d(X[i, :

本文详细介绍了卷积神经网络中的多输入通道和多输出通道的概念,以及1x1卷积的工作原理。1x1卷积通过调整通道数来控制模型复杂度,同时保持高和宽的尺寸不变,可以视为全连接层的特例。

6383

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



