如何解决Numpy:列表中所有向量的点积
假设我有一个包含 800 个向量的列表,每个向量由 1440 个标量组成。
List = [(1,2,3,...,1440),(1,...]
如何使用 python 和 numpy 以最快的方式计算每个向量与其他向量的点积?
解决方法
如果您想要对称矩阵,其中 x_ij
是 a_i
与 a_j
的点积,则:
a = np.array(List)
x = a @ a.T
,
您可以使用 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)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。