Skip to main content

farfield.unflatten_flux

farfield.unflatten_flux​

Unflattens a flux for a given expansion and Brillouin integration scheme.

Args:​

  • flux: The flux array, with shape (..., num_bz_kx, num_bz_ky, ... 2 * num_terms, num_sources).
  • expansion: The expansion used for the flux.
  • brillouin_grid_axes: The axes associated with the Brillouin zone grid.

Returns:​

  • None: The unflattened flux, with shape (..., num_kx, num_ky, 2, num_sources).