Global Average Pooling in Pytorch

[ywu36]

torch.mean works on one dimension instead of all three dimensions.