mirror of https://github.com/pyodide/pyodide.git
11 lines
319 B
Python
11 lines
319 B
Python
# http://stackoverflow.com/questions/7741878/how-to-apply-numpy-linalg-norm-to-each-row-of-a-matrix/7741976#7741976
|
|
# setup: import numpy as np ; N = 1000; x = np.random.rand(N,N)
|
|
# run: l2norm(x)
|
|
|
|
# pythran export l2norm(float64[][])
|
|
import numpy as np
|
|
|
|
|
|
def l2norm(x):
|
|
return np.sqrt(np.einsum('ij,ij->i', x, x))
|