public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-21 05:44:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

No need for np.repeat in normalize_along_axis

Browse source
This commit is contained in:
Grant Sanderson 2023-01-13 20:34:18 -08:00
parent 071e7f1a74
commit 42909a94ac
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
manimlib/utils/space_ops.py
View file

@ -193,8 +193,7 @@ def normalize_along_axis(
) -> np.ndarray: ) -> np.ndarray:
norms = np.sqrt((array * array).sum(axis)) norms = np.sqrt((array * array).sum(axis))
norms[norms == 0] = 1 norms[norms == 0] = 1
buffed_norms = np.repeat(norms, array.shape[axis]).reshape(array.shape) return (array.T / norms).T
return array / buffed_norms
def get_unit_normal( def get_unit_normal(