Numpy：列表中所有向量的点积答案

【问题标题】：Numpy: Dotproduct of all vectors in a listNumpy：列表中所有向量的点积
【发布时间】：2021-06-30 03:03:33
【问题描述】：

假设我有一个包含 800 个向量的列表，每个向量由 1440 个标量组成。

List = [(1, 2, 3, ... , 1440),
        (1, 2, 3, ... , 1440),
        (1, 2, 3, ... , 1440),
        ...]

如何使用 python 和 numpy 以最快的方式计算每个向量与其他向量的点积？

【问题讨论】：

【解决方案1】：

如果您想要x_ij 是a_i 与a_j 的点积的对称矩阵，则：

a = np.array(List)
x = a @ a.T

【讨论】：

【解决方案2】：

您可以使用np.triu_indices 或np.tril_indices 来避免计算矩阵的后半部分。这不会影响计算的复杂性，但可以为足够数量的长向量节省一些时间：

a = np.array(x)
n = a.shape[0]
r, c = np.triu_indices(n)
result = np.empty((n, n))
result[r, c] = result[c, r] = np.sum(a[r] * a[c], axis=1)

【讨论】：