使用其他函数计算的权重计算卷积

时间:2019-07-01 21:50:48

标签: pytorch

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

我正在实现本文的空间变体卷积:https://arxiv.org/abs/1804.00389

谢谢

1 个答案:

答案 0 :(得分:1)

实际上非常简单:torch.nn.functional.conv2d!