Elegant implementation of Spatial Pyramid Pooling layer?

3

Hi like in torch for using spatialpyramidpooling i can directly call (inn.SpatialPyramidPooling({8,8},{4,4},{2,2},{1,1})) can i do it similarily in some way in pytorch ?

What is the easiest way to do so.

Thanks