python - Pytorch:需要帮助实现自定义滑动窗口

Question

假设我有两个矩阵 A 和 B，都是 10x10。 现在我想要一个滑动窗口，3x3 来穿过 A 和 B 并且:

1. 计算以下结果:

然后获取该结果矩阵，求和，并将其保存在新矩阵内的位置e。

Answer 1

假设您正在应用加权和(即卷积)，您的内核相当于以下内容:

可以使用torch.nn.function.conv2d直接实现:

import torch.nn.functional as F

kernel = torch.Tensor([[-1,-2,-3],
                       [-4,sum([1,2,3,4,6,7,8,9]), -6],
                       [-7,-8,-9]])
kernel = kernel.reshape(1,1,3,3)

A = torch.randn(1,5,5)
B = torch.randn(1,5,5)
inputs = torch.stack([A,B])
F.conv2d(inputs, kernel, padding=1)