Hi,
I’m not sure I understand enough about what you describe  to comment on the NaNs.



 seanremy:

Ideally I would do something like np.convolve but there is not a pytorch version of that with autograd.


Would the functions for convolution  4 not work for this?
Best regards
Thomas