Sorry for pulling at this old thread, but can you explain how this is done…



 ptrblck:

 filter (create this as nn.Parameter if you want to train it)


Also, does implementing this whole in a forward function of an nn.Module class ensures proper flow of gradients and proper update of weights?