假設我有兩個大小為 (4000, 3) 的數組。我想以矢量化方式做的是計算從第一個數組的每個向量到第二個數組中的每個向量的 L1/Manhattan 距離，這樣我最終會得到一個 (4000, 4000) 數組。我目前的方法是基于將 (4000, 3) 拆分為 3 個單獨的 (4000, 1) 數組并進行廣播。但是，如果我有大小不同的初始矩陣，例如 (4000, 4) 或 (4000, 5)，則這種方法實際上不起作用。然后我的代碼會中斷，因為它假設有 3 個通道。因此，對于創建可以計算成對 L1 距離的廣義矢量化方法的任何幫助，我將不勝感激！
1 回答
料青山看我應如是
TA貢獻1772條經驗 獲得超8個贊
您可以使用廣播來完成整個事情（如果我理解您要正確執行的操作）。首先計算向量的成對差異（結果為 shape N,N,k），然后計算每個向量的絕對值之和。
N = 4000
k = 4
X = np.random.rand(N,k)
Y = np.random.rand(N,k)
Z = np.sum(np.abs(X[:,None]-Y[:]),axis=-1)
添加回答
舉報
0/150
提交
取消