【PyTorch】torch.einsumの挙動について - Qiita

はじめに Einsumは、様々な行列の演算ができます。通常、行列積や内積の計算では、行列の形に制約がありますが、Einsumは、添え字を使ってどんな形の行列でも計算ができてしまいます。 それゆえ、挙動が理解しにくいです。そこで、ここではfor文で実装して、挙動を分かりやすくしてみました。 torch.einsum 公式ドキュメント https://pytorch.org/docs/stable/generated/torch.einsum.html Einsumの実行結果 (4×3)のaと、(2×4)のbにEinsumを使ってみます。 import torch a = torch.tensor(range(1, 13)).reshape(4, 3) b = torch.tensor(range(1, 9)).reshape(2, 4) print(a) print(b) einsum1 =