python numpy逆_python-具有numpy的N * M * M张量的矢量化(部分)逆
所以我有一个张量的索引为a[n,i,j]的维度为(N,M,M)的张量,我想为N中的每个n求M*M方阵部分的值.例如,假设我有In[1]:a=np.arange(12)a.shape=(3,2,2)aOut[1]:array([[[0,1],[2,3]],[[4,5],[6,7]],[[8,9],[10,11]]])然后for循环反转将如下所示:In[2]:inv_a=np.zeros([3,2,2