Dear All,
I am very new to CNN and want to add my own developed pooling function to CNN instead of using traditional pooling methods like max-pooling and average pooling functions.  Can you suggest some way how to integrate my own pooling function to existing PyTorch code of CNN?
Thanks
Achyut