5
在numpy的,我們可以執行兩個向量a
和b
這樣之間的「外除了」: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
?
在numpy的,我們可以執行兩個向量a
和b
這樣之間的「外除了」: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
?
另一個快速替代這個操作是使用:
np.add.outer(a,b)
我不認爲'einsum'可以做到這一點(至少在沒有使用額外的數組)。我想不出一個比你寫的廣播純文本更快的方法,是不是太慢了?由於結果是對稱的,因此您可以編寫一個擴展,以儘可能少的總和。 – jorgeca