我想执行点(A,A.T),其中省略了某些索引:
In [6]: A = np.array([[1,2,3],[4,5,6],[7,8,9]])
In [6]: import numpy.ma as ma
In [7]: Ama = ma(A)
In [9]: Ama[0,0] = ma.masked
In [11]: ma.dot(Ama,Ama.T)
Out[11]:
masked_array(data =
[[13 28 43]
[28 77 122]
[43 122 194]],
mask =
[[False False False]
[False False False]
[False False False]],
fill_value = 999999)
最快的方法是什么?请注意,我想保留原始尺寸,即如果A是正方形,则点积将具有相同的尺寸。