QPMS
/
qpms
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix array indices 

Former-commit-id: f959fa55b001bd1ed3e175fed023ecad4c7008e7
This commit is contained in:
Marek Nečada 2017-02-16 00:59:21 +00:00
parent 2ff17ba650
commit ad563aaef2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
misc/dispersion-SVD.py
View File

@ -380,11 +380,11 @@ for trfile in os.scandir(translations_dir):
leftmatrixlist_TM = leftmatrixlist_s[np.ix_(np.arange(leftmatrixlist_s.shape[0]),TMč,TMč)] leftmatrixlist_TM = leftmatrixlist_s[np.ix_(np.arange(leftmatrixlist_s.shape[0]),TMč,TMč)]
svarr = np.linalg.svd(leftmatrixlist_TE, compute_uv=False) svarr = np.linalg.svd(leftmatrixlist_TE, compute_uv=False)
argsortlist = np.argsort(svarr, axis=-1)[...,:svn] argsortlist = np.argsort(svarr, axis=-1)[...,:svn]
minsvTElist[nnlist] = svarr[argsortlist] minsvTElist[nnlist] = svarr[...,argsortlist]
#minsvTElist[nnlist] = np.amin(np.linalg.svd(leftmatrixlist_TE, compute_uv=False), axis=-1) #minsvTElist[nnlist] = np.amin(np.linalg.svd(leftmatrixlist_TE, compute_uv=False), axis=-1)
svarr = np.linalg.svd(leftmatrixlist_TM, compute_uv=False) svarr = np.linalg.svd(leftmatrixlist_TM, compute_uv=False)
argsortlist = np.argsort(svarr, axis=-1)[...,:svn] argsortlist = np.argsort(svarr, axis=-1)[...,:svn]
minsvTMlist[nnlist] = svarr[argsortlist] minsvTMlist[nnlist] = svarr[...,argsortlist]
#minsvTMlist[nnlist] = np.amin(np.linalg.svd(leftmatrixlist_TM, compute_uv=False), axis=-1) #minsvTMlist[nnlist] = np.amin(np.linalg.svd(leftmatrixlist_TM, compute_uv=False), axis=-1)
minsvTMlistlist.append(minsvTMlist) minsvTMlistlist.append(minsvTMlist)
minsvTElistlist.append(minsvTElist) minsvTElistlist.append(minsvTElist)