Hi,
I think the problem is just that you cannot use the out= keyword argument within the autograd engine.
In the example above, since you don’t require grads, this does not use the autograd and so works well.
I guess when you use the functional version of conv, you don’t require grads and so it works as well.
When using the nn version, the conv parameters require grads by default and so will fail.