我有兩個 numpy 矩陣，A 和 B：A = [[0,-1],     [-2,5],     [2,6]]B = [[5,3],     [2,4]]Dimension of A: N * KDimension of B: K * K我想要做的是對A 的每一行和整個 B進行 marix 乘法，然后再將它與A 的每一行相乘。然后總結(jié)一下。這是：N = A.shape[0]sum = 0for i in range(N):    s = A[i] @ B @ A[i]    print(s)    sum += sreturn sum這印470224而4 + 70 + 224 = 298這正是我想要的。在沒有 for 循環(huán)的情況下，是否有任何優(yōu)雅或 Pythonic 的方式來實現(xiàn)這一點？謝謝