[numpy] 行列・ベクトルのドット積(内積)の計算 [Python]

行列・ベクトルのドット積の計算

numpyのdot関数か、@演算子を使うことで求められます。

import numpy as np
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
arr1.dot(arr2) # 32

import numpy as np
arr1 = np.array([[1, 2],
                 [3, 4]])
arr2 = np.array([[2, 1],
                 [1, 2]])
arr1.dot(arr2)

OUT

array([[ 4,  5],
       [10, 11]])

import numpy as np
arr1 = np.array([[1, 2],
                 [3, 4]])
arr2 = np.array([1, 2])
arr1.dot(arr2)

OUT

array([ 5, 11])