在3D矩阵上广播linalg.inv

Question

在下面的例子中，有一个大小为（4，3，3）的3d矩阵。如何在numpy中计算没有循环的3*3矩阵中的4个矩阵的pinv？

dt = np.dtype(np.float32)
a=[[[12,3,1],
   [2,4,1],
   [2,4,2],],
   [[12,3,3],
   [2,4,4],
   [2,4,5],],
   [[12,3,6],
   [2,4,5],
   [2,4,4],],
   [[12,3,3],
   [2,4,5],
   [2,4,6]]]
a=np.asarray(a,dtype=dt)
print(a.shape)
apinv=np.zeros((4,3,3))
print(np.linalg.pinv(a[0,:,:]).shape)
for i in range(4):
   apinv[i,:,:]=np.linalg.pinv(a[i,:,:])

请注意，linalg.iv通过矩阵进行广播，如下所述：

print(np.linalg.inv(a).shape)

但它对pinv的作用不同。

Answer 1

一个解决方案是使用映射和lambda函数：

apinv = map(lambda n: np.linalg.pinv(n), a)
apinv = np.asarray(apinv,dtype=dt)