2013-07-11 25 views
5

在numpy的,我們可以執行兩個向量ab這樣之間的「外除了」:numpy einsum()可以執行外部添加嗎?

a=np.c_[1,2,3] 
b=np.c_[4,5,6] 
result=a+b.T # alternatively this can be a.T+b 

是否有可能使用einsum也作出了同樣的計算?任何其他快速替代品?如果a等於b

+3

我不認爲'einsum'可以做到這一點(至少在沒有使用額外的數組)。我想不出一個比你寫的廣播純文本更快的方法,是不是太慢了?由於結果是對稱的,因此您可以編寫一個擴展,以儘可能少的總和。 – jorgeca

回答

9

另一個快速替代這個操作是使用:

np.add.outer(a,b) 
相關問題