给定多个矩阵的列表，我想计算每个矩阵的迹。
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)
关于python - 跟踪 numpy 数组中的每个矩阵，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/55973035/