标签: pytorch
我想执行一次conv2d但这是一种特殊的操作,因为我为每个样本计算了函数中conv2d层的权重。我可以简单地为每个样品分配给定层的权重吗?
我正在实现本文的空间变体卷积:https://arxiv.org/abs/1804.00389
谢谢
答案 0 :(得分:1)
实际上非常简单:torch.nn.functional.conv2d!