卷积和上采样过程中的特征数量计算

卷积过程中的特征数量计算

参考链接

链接: https://blog.csdn.net/qq_34392457/article/details/103949063

// An highlighted block
            nn.Conv3d(256, out_channel,kernel_size=(1,2,2), stride=2)
            nn.Upsample(scale_factor=(16, 32, 32), mode='trilinear'),
            nn.Softmax(dim =1)

输入：（1, 256, 3, 32, 32）
经过上面代码第一行(卷积)后
输出：（1, 256, 3, 16, 16）
经过第二行（上采样）后
计算过程结合scale_factor，316, 1632, 16*32
输出：（1, 2, 48, 512, 512）

记录一下！