Global Average Pooling Pytorch