Dot product two 4D Numpy array

Question

I have a 4D Numpy array of size [15, 2, 320, 320]. In other words, each element of the [320 x 320] matrix is a matrix of size [15 x 2]. Now, I would like to compute the dot