size为(256,32,36,256)的tensor,输入卷积层nn.Conv2d(32,64,(3,3),stride=(2,2),padding=(1,1)),输出的size是什么?
时间: 2023-10-14 09:07:07 浏览: 132
TensorFlow tf.nn.conv2d实现卷积的方式
5星 · 资源好评率100%
根据卷积层的计算公式,输出的size应该是(256, 64, 18, 128)。其中,256表示batch size,64表示输出通道数,18表示输出特征图的高度,128表示输出特征图的宽度。计算方法如下:
- 高度维度:(32 + 2x1 - 3) / 2 + 1 = 18
- 宽度维度:(36 + 2x1 - 3) / 2 + 1 = 128
因此,输出的size为(256, 64, 18, 128)。
阅读全文