a ij @sRdZddlmZddlmZddlmZddlmZejZd dddd d d Z d S)zC`linalg` compat module for interoperability between tf, jax, numpy.) annotations)compat) numpy_utils) FloatArrayzFloatArray['*d']int)xaxisreturncCs|tj||ddS)z&Normalize the vector to the unit norm.T)r Zkeepdims)rZnorm)rr r N/home/ghrups/robot-bench/.venv/lib/python3.9/site-packages/etils/enp/linalg.py normalizesr N)r) __doc__ __future__rZ etils.enprrZetils.enp.typingrZlazyr r r r r s