jan/meanas
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

cleaner way to ravel

Browse Source
This commit is contained in:
Jan Petykiewicz 2023-05-22 09:53:30 -07:00
parent 7cf90fe0de
commit fd1a83b5b9
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
meanas/fdfd/bloch.py
View File

@ -257,7 +257,9 @@ def maxwell_operator(
# transform back to mn # transform back to mn
h_m = numpy.sum(h_xyz * m, axis=3) h_m = numpy.sum(h_xyz * m, axis=3)
h_n = numpy.sum(h_xyz * n, axis=3) h_n = numpy.sum(h_xyz * n, axis=3)
return numpy.hstack((h_m.ravel(), h_n.ravel()))
h = numpy.concatenate((h_m, h_n), axis=None, out=h) # ravel and merge
return h
return operator return operator
@ -412,7 +414,8 @@ def inverse_maxwell_operator_approx(
h_m = numpy.sum(d_xyz * n, axis=3, keepdims=True) / +k_mag h_m = numpy.sum(d_xyz * n, axis=3, keepdims=True) / +k_mag
h_n = numpy.sum(d_xyz * m, axis=3, keepdims=True) / -k_mag h_n = numpy.sum(d_xyz * m, axis=3, keepdims=True) / -k_mag
return numpy.hstack((h_m.ravel(), h_n.ravel())) h = numpy.concatenate((h_m, h_n), axis=None, out=h)
return h
return operator return operator