numpyの行列乗算行単位

Question

私は

a = np.array([[1, 2, 3], [2, 3, 4], [3, 4, 5], [4, 5, 6]]) 

のように、1つの行列を持っていると私は各要素は、それ自体での行の行列積である新しい行列を取得したい：

np.array([ 
    np.dot(np.array([a[0]]).T, np.array([a[0]])), 
    np.dot(np.array([a[1]]).T, np.array([a[1]])), 
    np.dot(np.array([a[2]]).T, np.array([a[2]])), 
    np.dot(np.array([a[3]]).T, np.array([a[3]])), 
]) 

は、各要素が3x3の行列である4x4行列になります。 この後、0軸を合計して新しい3x3行列を得ることができます。

ループを使用する以外はこれを実装する上手な方法はありますか？整列した第1の軸を保つ2つ目の間の外積を実行するための

Answer 1

使用NumPy broadcasting - 私は何かが欠けていないが、可能性があります

np.einsum('ij,ik->ijk',a,a) 

Answer 2

- np.einsumで

a[:,:,None]*a[:,None,:] # or a[...,None]*a[:,None] 

は、に変換しますこれだけの行列乗算？

>>> a.T @ a 
array([[30, 40, 50], 
     [40, 54, 68], 
     [50, 68, 86]]) 

>>> np.sum(np.array([ 
     np.dot(np.array([a[0]]).T, np.array([a[0]])), 
     np.dot(np.array([a[1]]).T, np.array([a[1]])), 
     np.dot(np.array([a[2]]).T, np.array([a[2]])), 
     np.dot(np.array([a[3]]).T, np.array([a[3]])), 
    ]), axis=0) 
array([[30, 40, 50], 
     [40, 54, 68], 
     [50, 68, 86]])