[英]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.