pytorch问题集合

根据kernel size,stride和padding计算卷积后的尺寸

对于PyTorch中的1维卷积层nn.Conv1d,输出序列长度可以根据以下公式计算:假设:- 输入序列长度:L_in 
- 卷积核大小:K
- 步长:S
- 填充:P
则输出序列长度为:

python
L_out = (L_in + 2 * P - K) // S + 1

这里://表示地板除(向下取整)。举例:- 输入序列长度L_in = 28
- 卷积核大小K = 5
- 步长S = 2
- 填充P = 1
则:

L_out = (L_in + 2 * P - K) // S + 1
     = (28 + 2 * 1 - 5) // 2 + 1
     = 13