Torch.nn.functional.normalize