def _get_diagIDs(K):
if K in diagIDsDict:
return diagIDsDict[K]
else:
diagIDs = np.diag_indices(K)
diagIDsDict[K] = diagIDs
return diagIDs
微信
公众号
扫码关注公众号