繁体   English   中英

是否有用于计算矩阵列或向量列表中所有对之间的所有成对点积的函数?

[英]Is there a function for calculating all the pairwise dot products of the columns of a matrix, or between all pairs in a list of vectors?

说你有n。 任意(但相等)长度为 m 的向量。 是否有(numpy?)函数或快速方法来计算这些 n 之间的所有成对点积。 向量?

我最初的想法是你可以计算 A T A 并取上三角部分,但我不确定矩阵乘法是否很慢/计算密集。 有没有更快/有效的方法? 或者我应该只使用 for 循环为所有对组合定义一个函数?

根据@Brenila 的评论,使用tensordot

np.tensordot(arr, arr, axes=(0,0))

对于n = arr.shape[-1] (n, n)结果形状为(n, n)

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM