numpy在矩阵数组上应用函数

Question

假设我有一个2D矩阵数组（即3D数组）

a = np.random.rand(10,2,2)

现在，我想对这10个2x2矩阵应用一个函数。例如

def test(i):
    return i.shape
b  = np.array([test(i) for i in a])

因此，b将是10个元组(2,2)的数组。现在，没有显式循环，有什么numpy方法可以做到这一点？我尝试过apply_along_axis，但只在一个轴上应用了该功能。