Pythonでrank 4以上のテンソル積の求め方

Question

###前提・実現したいこと
Python 2.7でRank4以上の計算をしたいです

###試したこと

import numpy as np
a = np.arange(0, 12).reshape(2,2,2,3)
b = np.arange(0, 12).reshape(2,2,3,2)
np.matmul(a,b)

matmul, einsum, dotなどを試しましたがどれもうまく行きません
前提として(a, b, c, d)と(b, c, d, e)で積が定義されていると思っていたのですが
間違いなのでしょうか…？

Answer 1

numpy.tensordot() を使ってください。

a = np.empty((2,2,2,3))
b = np.empty((2,2,3,2))

np.tensordot(a,b, axes=0).shape
(2, 2, 2, 3, 2, 2, 3, 2)

Comments

[heroherotail]

axesの部分が必要だったのですね
大変助かりました！