global-average-pooling-pytorch