python之跟踪 numpy 数组中的每个矩阵

给定多个矩阵的列表，我想计算每个矩阵的迹。

np.asarray([[[1,2,3], 
             [4,5,6], 
             [7,8,9]], 
           [[10,11,12], 
            [13,14,15], 
            [16,17,18]]]) 

结果应该是:

np.asarray([15, 42]) 

请您参考如下方法:

在 np.trace 中使用 axis1=1 和 axis2=2 参数。

np.trace(x, axis1=1, axis2=2)