这篇博客探讨了 Numpy 中 `dot`、`@` 和 `np.dot` 函数进行矩阵乘法时的性能和用法。根据测试，它们的性能几乎相同，但在某些情况下，`matmul` 或 `@` 符号被推荐使用。Numpy 文档建议，对于 1D 和 2D 数组，这三个操作都是等价的，但如果涉及 0D 或 N-D 数组，推荐使用 `numpy.multiply` 或简单的乘法运算符 `*`。
Is there any difference? If not, what is preferred by convention?
The performance seems to be almost the same.
a=np.random.rand(1000,1000)
b=np.random.rand(1000,1000)
%timeit a.dot(b) #14.3 ms ± 374 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)
%timeit np.dot(a,b) #14.7 ms ± 315 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)
%timeit a @ b #15.1 ms ± 779 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)
解决方案
They are all basically doing the same thing. In terms of timing, based on Numpy's documentation here:
If both a and b are 1-D arrays, it is inner product of vectors
(without complex conjugation).
If both a and b are 2-D arrays, it is matrix multiplication, but
using matmul or a @ b is preferred.
If either a or b is 0-D (scalar), it is equivalent to multiply and
using numpy.multiply(a, b) or a * b is preferred.
If a is an N-D array and b is a 1-D array, it is a sum product over
the last axis of a and b.
1万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
