示例代码:
import numpy as np
import math
import time
x=np.ones((2000,2000))
start = time.time()
print(np.linalg.norm(x, 2))
end = time.time()
print("time 1: " + str(end - start))
start = time.time()
print(math.sqrt(np.sum(x*x)))
end = time.time()
print("time 2: " + str(end - start))
(在我的机器上)输出为:
1999.999999999991
time 1: 3.216777801513672
2000.0
time 2: 0.015042781829833984
它表明np.linalg.norm()花费了3s以上的时间来解决它,而直接解决方案只花费了0.01s。为什么np.linalg.norm()这么慢?
答案 0 :(得分:2)
np.linalg.norm(x, 2)
计算2范数,取最大的奇异值
math.sqrt(np.sum(x*x))
计算frobenius范数
这些操作是不同的,因此花费不同的时间也就不足为奇了。 What is the difference between the Frobenius norm and the 2-norm of a matrix?可能对math.SO感兴趣。
答案 1 :(得分:0)
可比的是:
In [10]: %timeit sum(x*x,axis=1)**.5
36.4 ms ± 6.11 ms per loop (mean ± std. dev. of 7 runs, 10 loops each)
In [11]: %timeit norm(x,axis=1)
32.3 ms ± 3.94 ms per loop (mean ± std. dev. of 7 runs, 10 loops each)
np.linalg.norm(x, 2)
和sum(x*x)**.5
都不是同一件事。