特征图通道数的改变 .repeat()函数或者1X1卷积通道改变法

.repeat()函数


out = self.sigmoid(self.conv1(out))  # out.shape = 8,1,32,32
out = out.repeat(1, 64, 1, 1)  # repeat后是 一一对应相乘积 此时的out.shape = 8,64,32,32
        

通道数的改变也可以通过1X1卷积直接改变

你可能感兴趣的:(深度学习)