Are there any plans to add a pytorch equivalent of the mxnet correlation layer? Right now I’m going to try implementing a custom layer, similar to this but I think it would be nice to have out-of-the-box.
Thanks, NVIDIA.
1 Like
would be nice to add the released PyTorch implementations of correlation and resample2d custom layers to the standard PyTorch library
3 Likes