size为(256,64,18,128)的tensor,输入卷积层 nn.Conv2d(64,128,(6,6),stride=(3,3)),,输出的size是什么?
时间: 2023-10-27 12:06:29 浏览: 93
根据卷积层的定义,输出的size可以通过以下公式计算:
$$
\text{Output size} = \left\lfloor\frac{\text{Input size} - \text{Kernel size}}{\text{Stride}}\right\rfloor + 1
$$
其中,$\lfloor\rfloor$表示向下取整。
因此,对于输入的size为$(256, 64, 18, 128)$,进行卷积核大小为$(6, 6)$,步长为$(3, 3)$的卷积操作,输出的size为:
$$
\begin{aligned}
\text{Output size} &= \left\lfloor\frac{\text{Input size} - \text{Kernel size}}{\text{Stride}}\right\rfloor + 1 \\
&= \left\lfloor\frac{(256, 64, 18, 128) - (6, 6)}{(3, 3)}\right\rfloor + 1 \\
&= \left\lfloor(83.33, 19.33, 4, 41.33)\right\rfloor + 1 \\
&= (83, 19, 4, 41) + 1 \\
&= (84, 20, 5, 42)
\end{aligned}
$$
因此,输出的size为$(84, 20, 5, 42)$。
阅读全文