Pythonでforループをベクトル演算に

Question

pythonはforループを回すより、リスト内包表記やベクトル演算に置き換えた方が早いと言われている様なのですが、下記の計算をベクトル演算に置き換えることは可能でしょうか？

import numpy as np

m=np.zeros((10,10))
n=np.ones((10,10))
for i in range(10):
  for j in range(10):
    m[i,j]=np.dot(n[:,i],n[:,j])

10行10列の行列nのi列とj列の内積を、行列mに格納したいということです。
リスト内包表記やmap関数でやるしかないのでしょうか？

Comments

[t_obara]

実際に上記のコードかそれに類似するコードを実行して試してみたのでしょうか？

[yamane]

上記の配列の次元が大きいもの（と、そのリスト内包表記）を書いて思うような速度が得られなかったので質問させていただきました。

Answer 1

(n[:,np.newaxis,:]*n.T[np.newaxis,:,:]).sum(axis=2)

Comments

[yamane]

ありがとうございました。できるんですね。newaxisは初見でした。