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

be explicit about stack axis

Browse Source
This commit is contained in:
Jan Petykiewicz 2022-10-04 14:33:36 -07:00
parent 71d6ceec1e
commit ec9a6edc85
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
meanas/fdfd/bloch.py
View File

@ -198,9 +198,9 @@ def maxwell_operator(
shape = epsilon[0].shape + (1,) shape = epsilon[0].shape + (1,)
k_mag, m, n = generate_kmn(k0, G_matrix, shape) k_mag, m, n = generate_kmn(k0, G_matrix, shape)
epsilon = numpy.stack(epsilon, 3) epsilon = numpy.stack(epsilon, axis=3)
if mu is not None: if mu is not None:
mu = numpy.stack(mu, 3) mu = numpy.stack(mu, axis=3)
def operator(h: NDArray[numpy.float64]) -> NDArray[numpy.float64]: def operator(h: NDArray[numpy.float64]) -> NDArray[numpy.float64]:
""" """
@ -272,7 +272,7 @@ def hmn_2_exyz(
Function for converting `h_mn` into `E_xyz` Function for converting `h_mn` into `E_xyz`
""" """
shape = epsilon[0].shape + (1,) shape = epsilon[0].shape + (1,)
epsilon = numpy.stack(epsilon, 3) epsilon = numpy.stack(epsilon, axis=3)
k_mag, m, n = generate_kmn(k0, G_matrix, shape) k_mag, m, n = generate_kmn(k0, G_matrix, shape)
@ -350,12 +350,12 @@ def inverse_maxwell_operator_approx(
Function which applies the approximate inverse of the maxwell operator to `h_mn`. Function which applies the approximate inverse of the maxwell operator to `h_mn`.
""" """
shape = epsilon[0].shape + (1,) shape = epsilon[0].shape + (1,)
epsilon = numpy.stack(epsilon, 3) epsilon = numpy.stack(epsilon, axis=3)
k_mag, m, n = generate_kmn(k0, G_matrix, shape) k_mag, m, n = generate_kmn(k0, G_matrix, shape)
if mu is not None: if mu is not None:
mu = numpy.stack(mu, 3) mu = numpy.stack(mu, axis=3)
def operator(h: NDArray[numpy.float64]) -> NDArray[numpy.float64]: def operator(h: NDArray[numpy.float64]) -> NDArray[numpy.float64]:
""" """