diff --git a/sfs/util.py b/sfs/util.py index de2ead57..ac1e1456 100644 --- a/sfs/util.py +++ b/sfs/util.py @@ -338,6 +338,12 @@ def normalize_vector(x): return x / np.linalg.norm(x) +def normalize_rows(x): + """Normalize a list of vectors.""" + x = asarray_of_rows(x) + return x / np.linalg.norm(x, axis=1, keepdims=True) + + def db(x, *, power=False): """Convert *x* to decibel.