Pytorch 使用由另一个函数计算的权重计算卷积

我想执行conv2d，但这是一个特殊的操作，因为我在函数中为每个样本计算conv2d层的权重。我可以简单地为每个样本分配给定层的权重吗

我正在实现本文的空间变异卷积：

---

谢谢

其实很简单：torch.nn.functional.conv2d