Custom pooling layer in Pytorch

Job ID: 31987360

Budget: $10 – $11 CAD

I need a custom pooling layer which must work in ANY model.
The calculation part should be easily customizable.
Back and forward propagation should be implemented.
config and size of input and output should be calculated automatically as if you are requested to implement exactly MaxPooling2D.
budget is $15.
24-48 hours
Related categories: Pytorch Deep Learning