为什么np.linalg.norm(x,2)比直接解决它要慢?

时间:2018-10-14 15:09:13

标签: python performance numpy norm

示例代码:

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()这么慢?

2 个答案:

答案 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都不是同一件事。