Есть ли элегантный, тупой способ поэтапного применения точечного произведения? Или как можно перевести приведенный ниже код в более приятную версию?

m0 # shape (5, 3, 2, 2)
m1 # shape (5,    2, 2)
r = np.empty((5, 3, 2, 2))
for i in range(5):
    for j in range(3):
        r[i, j] = np.dot(m0[i, j], m1[i])

Заранее спасибо!