Global Pooling Pytorch