Pytorch Global Average Pooling 3d