Average pooling by desired output shape

Question

I have a logic question.

I\'m trying to create some sort of version of PyTorch\'s nn.functional.avg_pool function with a slight modification: I want the