numpy行列ベクトル乗算

| | | | | | | | | |

サイズ(nxn)*(nx 1)の2つのnumpy配列を乗算すると、サイズ(nxn)の行列が得られます。通常の行列乗算規則に従うと、(nx 1)ベクトルが期待されますが、PythonのNumpyモジュールでこれがどのように行われるかについての情報を見つけることができません。

問題は私が望んでいないことですプログラムの速度を維持するために手動で実装します。

サンプルコードを以下に示します。

  a = np.array([[5、1、3]、 [1、1、1]、[1、2、1]])b = np.array([1、2、3])print a * b>> [[5 2 9] [1 2 3] [1 4 3]]  

必要なもの:

  print a * b>&gt ; [16 6 8]