Вычислить свертку с весами, которые вычисляются другой функцией

Question

Я хотел бы выполнить conv2d, но особый, потому что я вычисляю веса слоя conv2d в функции для каждой выборки. Могу ли я просто назначить веса данного слоя для каждого образца?

Я реализую пространственно-вариантную свертку этой статьи: https://arxiv.org/abs/1804.00389

Спасибо

Answer 1

Это на самом деле очень просто: torch.nn.functional.conv2d!