旅行好きなソフトエンジニアの備忘録

プログラミングや技術関連のメモを始めました

【Python】 行列積の演算子

行列積はNumPyのdotメソッドを利用しますが、Python3.5/NumPy1.10からは@演算子でも行列積を表現できると知ったのでメモします。

import numpy as np

A = np.array([[1, 2], [3, 4]])
B = np.array([[5, 6], [7, 8]])
# a_dot_bとa_at_bは同じ結果になります
a_dot_b = np.dot(A, B)
a_at_b = A@B

科学技術計算のためのPython入門 ――開発基礎、必須ライブラリ、高速化

科学技術計算のためのPython入門 ――開発基礎、必須ライブラリ、高速化