有没有办法矢量化linalg.expm

Question

跟进this question 我是python的新手，我正在尝试计算指数 产品矩阵 - 使用矢量化的标量（如果可能）

我做了什么：

n=10
t_ = np.arange(1,n+1)*5*np.pi/n
a_11,a_12, a_21, a_22=0,1,-1,-1
x_0,v_0=1,1
A = np.array([[a_11,a_12], [a_21, a_22]])  
A_ = np.array([A  for k in  range (1,n+1,1)])

X_0 = np.array([[x_0],[v_0]]) # build X_0
print A

x_=scipy.linalg.expm(t_[:,None,None]*A[None,:,:])*X_0

我在linalg.expm中收到以下错误：

ValueError: expected a square matrix

非常感谢任何帮助。