How can I take a np.einsum along a specific dimension?

Question

I have 2 complex matricies, each of the shape (2,4,4,4,3). I would like to implement the following code without a for loop:

ce2 = np.random.rand(2,4,4,4,3)+np.ran