簡體   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