I would like to convert a NumPy array to a unit vector. More specifically, I am looking for an equivalent version of this normalisation function:

```
def normalize(v):
norm = np.linalg.norm(v)
if norm == 0:
return v
return v / norm
```

This function handles the situation where vector `v`

has the norm value of 0.

Is there any similar functions provided in `sklearn`

or `numpy`

?