I have as an output of a convolutional network a tensor of shape [1, 20, 64, 64]. I want to normalize all feature maps to a range of [0, 1]. I found out, that I can get all the means with means = torch.mean(features, (2, 3)), but I don’t know how to proceed further. Any help?