Torch Functional Normalize