diff options
Diffstat (limited to 'numpy/linalg/linalg.py')
| -rw-r--r-- | numpy/linalg/linalg.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/numpy/linalg/linalg.py b/numpy/linalg/linalg.py index 36c5eb85c..7ce854197 100644 --- a/numpy/linalg/linalg.py +++ b/numpy/linalg/linalg.py @@ -24,7 +24,8 @@ from numpy.core import ( add, multiply, sqrt, fastCopyAndTranspose, sum, isfinite, finfo, errstate, geterrobj, moveaxis, amin, amax, product, abs, atleast_2d, intp, asanyarray, object_, matmul, - swapaxes, divide, count_nonzero, isnan, sign, argsort, sort + swapaxes, divide, count_nonzero, isnan, sign, argsort, sort, + reciprocal ) from numpy.core.multiarray import normalize_axis_index from numpy.core.overrides import set_module @@ -2561,7 +2562,7 @@ def norm(x, ord=None, axis=None, keepdims=False): absx = abs(x) absx **= ord ret = add.reduce(absx, axis=axis, keepdims=keepdims) - ret **= (1 / ord) + ret **= reciprocal(ord, dtype=ret.dtype) return ret elif len(axis) == 2: row_axis, col_axis = axis |