Torch Nn Functional Normalize