Pytorch normalize tensor between 0 and 1.