假設(shè)我有兩個(gè)大小為 (4000, 3) 的數(shù)組。我想以矢量化方式做的是計(jì)算從第一個(gè)數(shù)組的每個(gè)向量到第二個(gè)數(shù)組中的每個(gè)向量的 L1/Manhattan 距離，這樣我最終會(huì)得到一個(gè) (4000, 4000) 數(shù)組。我目前的方法是基于將 (4000, 3) 拆分為 3 個(gè)單獨(dú)的 (4000, 1) 數(shù)組并進(jìn)行廣播。但是，如果我有大小不同的初始矩陣，例如 (4000, 4) 或 (4000, 5)，則這種方法實(shí)際上不起作用。然后我的代碼會(huì)中斷，因?yàn)樗僭O(shè)有 3 個(gè)通道。因此，對(duì)于創(chuàng)建可以計(jì)算成對(duì) L1 距離的廣義矢量化方法的任何幫助，我將不勝感激！
1 回答
料青山看我應(yīng)如是
TA貢獻(xiàn)1772條經(jīng)驗(yàn) 獲得超8個(gè)贊
您可以使用廣播來(lái)完成整個(gè)事情（如果我理解您要正確執(zhí)行的操作）。首先計(jì)算向量的成對(duì)差異（結(jié)果為 shape N,N,k），然后計(jì)算每個(gè)向量的絕對(duì)值之和。
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)
添加回答
舉報(bào)
0/150
提交
取消