python - 一维 PyTorch 圆形填充

Question

对于卷积，我想在一个维度上应用圆形填充，在所有其他维度上应用零填充。我该怎么做？

对于卷积，有 28 个 channel ，并且数据在球形箱中描述。半径时间有 20 个箱子，极地时间有 20 个箱子，倾角有 20 个箱子。 圆形填充应仅适用于倾斜。

小例子

# Example:
x = torch.tensor([[1,2,3],
                  [4,5,6],
                  [7,8,9]])
y = sphere_pad(x, pad=(0, 1))

# y is now tensor([[3, 1, 2, 3, 1],
#                  [6, 4, 5, 6, 4],
#                  [9, 7, 8, 9, 7]])

我已经尝试申请了

def sphere_pad(x, pad=(1,1)):
    return x.repeat(*x.shape)[
        (x.shape[0]-pad[0]):(2*x.shape[0]+pad[0]), 
        (x.shape[1]-pad[1]):(2*x.shape[1]+pad[1])]

然后应用具有正常零填充的卷积(并且在最后一个维度中没有填充)。 这适用于一个小示例，但此方法超出了实际问题大小的 GPU 内存。 还有其他方法吗？

Answer 1

使用 numpy，您可以执行 wrap 填充，以便数组沿第二个轴包裹:

np.pad(x, ((0,0),(1,1)), mode='wrap')
array([[3, 1, 2, 3, 1],
       [6, 4, 5, 6, 4],
       [9, 7, 8, 9, 7]])