Function normalize

humanoidverse/utils/torch_utils.py:46–47

(x, eps: float = 1e-9)

@torch.jit.script
def normalize(x, eps: float = 1e-9):
    return x / x.norm(p=2, dim=-1).clamp(min=eps, max=None).unsqueeze(-1)


@torch.jit.script