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

use keepdims instead of readding dims

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

8
meanas/fdfd/bloch.py
View File

@ -241,8 +241,8 @@ def maxwell_operator(
e_xyz = fftn(ifftn(d_xyz, axes=range(3)) / epsilon, axes=range(3)) e_xyz = fftn(ifftn(d_xyz, axes=range(3)) / epsilon, axes=range(3))
# cross product and transform into mn basis # cross product and transform into mn basis
b_m = numpy.sum(e_xyz * n, axis=3)[:, :, :, None] * -k_mag b_m = numpy.sum(e_xyz * n, axis=3, keepdims=True) * -k_mag
b_n = numpy.sum(e_xyz * m, axis=3)[:, :, :, None] * +k_mag b_n = numpy.sum(e_xyz * m, axis=3, keepdims=True) * +k_mag
if mu is None: if mu is None:
h_m, h_n = b_m, b_n h_m, h_n = b_m, b_n
@ -409,8 +409,8 @@ def inverse_maxwell_operator_approx(
d_xyz = fftn(ifftn(e_xyz, axes=range(3)) * epsilon, axes=range(3)) d_xyz = fftn(ifftn(e_xyz, axes=range(3)) * epsilon, axes=range(3))
# cross product and transform into mn basis crossinv_t2c # cross product and transform into mn basis crossinv_t2c
h_m = numpy.sum(d_xyz * n, axis=3)[:, :, :, None] / +k_mag h_m = numpy.sum(d_xyz * n, axis=3, keepdims=True) / +k_mag
h_n = numpy.sum(d_xyz * m, axis=3)[:, :, :, None] / -k_mag h_n = numpy.sum(d_xyz * m, axis=3, keepdims=True) / -k_mag
return numpy.hstack((h_m.ravel(), h_n.ravel())) return numpy.hstack((h_m.ravel(), h_n.ravel()))