如果n-1/ 2不是整数值，如何在PyTorch中执行“相同”填充

如何解决如果n-1/ 2不是整数值，如何在PyTorch中执行“相同”填充

我正在尝试重建用张量流编写的神经网络。对于卷积层，它们仅使用padding='SAME'。在pytorch中不存在。我知道，我可以使用p = (n - 1) / 2来计算stride = 1的填充。但是，如果这不会导致整数值怎么办？在我的情况下，n为4，我一直想获得相同的填充。

解决方法

使用math.floor函数将向下舍入到最接近的整数或math.ceil函数将上舍入到最接近的整数：

import math

# for flooring
p = math.floor((n - 1) / 2))

# for ceiling
p = math.ceil((n - 1) / 2))

例如，默认情况下，pytorch对MaxPool层使用地板。因此，我认为flooring是一个很好的起点。