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

Fix normalize_along_axis

Browse source
This commit is contained in:
Grant Sanderson 2022-12-22 11:52:55 -07:00
parent 5af4b9cc65
commit a4d47f64b0
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

@ -178,8 +178,7 @@ def normalize_along_axis(
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) buffed_norms = np.repeat(norms, array.shape[axis]).reshape(array.shape)
array /= buffed_norms return array / buffed_norms
return array
def get_unit_normal( def get_unit_normal(