7

「np.einsum」のドキュメントで私が理解したのは、順列文字列がベクトルの軸の順列を与えるということです。これは、次の実験で確認されています。

>>> M = np.arange(24).reshape(2,3,4)
>>> M.shape
(2, 3, 4)
>>> np.einsum('ijk', M).shape
(2, 3, 4)
>>> np.einsum('ikj', M).shape
(2, 4, 3)
>>> np.einsum('jik', M).shape
(3, 2, 4)

しかし、これは理解できません:

>>> np.einsum('kij', M).shape
(3, 4, 2)

代わりに (4, 2, 3) を期待します...私の理解の何が問題になっていますか?

4

1 に答える 1