numpy non canonical dot product in higher dimension

Question

I\'m trying to vectorize a loop with NumPy but I\'m stuck I have a matrix A of shape (NN,NN) I define the A-dot product by