Pytorch Global Average Pooling