使用 numpy 在特定轴上行列式
Determinant over a specific axis using numpy
假设我有一个形状为 (j,d,d) 的 numpy 数组 A，我想获得一个形状为 j 的数组，其中每个条目对应于每个 (d,d) 数组的行列式。
我尝试使用 np.apply_along_axis(np.linalg.det(A), axis=0)，但 np.apply_along_axis 似乎只适用于一维切片。
是否有仅使用 numpy 的有效方法？
np.linalg.det 已经可以对任意形状的数组执行此操作，只要最后两个维度是正方形即可。您可以查看文档 here.
假设我有一个形状为 (j,d,d) 的 numpy 数组 A，我想获得一个形状为 j 的数组，其中每个条目对应于每个 (d,d) 数组的行列式。
我尝试使用 np.apply_along_axis(np.linalg.det(A), axis=0)，但 np.apply_along_axis 似乎只适用于一维切片。
是否有仅使用 numpy 的有效方法？
np.linalg.det 已经可以对任意形状的数组执行此操作，只要最后两个维度是正方形即可。您可以查看文档 here.