tefield¶

class tefield.TE_grid(coords: jax.Array | List[jax.Array] | Tuple[jax.Array], coords_field: jax.Array | List[jax.Array] | Tuple[jax.Array])[source]¶

Bases: object

grid TE field model. See Synax paper for more details.

Parameters:

coords (Union[jax.Array,List[jax.Array],Tuple[jax.Array]]) – coordinates of all integration points. Should be of size (3,…), for example coords[0] is the x-coordinates.
coords_field (Union[jax.Array,List[jax.Array],Tuple[jax.Array]]) – coords[i] is the 1D vector of coordinates along i-th axis. Since the grid is a regular 3D grid, 1D vectors are sufficient to represents the coordinates.

Returns:

A instance of grid TE field generator.

TE_field(te_field_grid)¶

Calculate grid TE-field at all positions specified by coords.

Parameters:: TE_field_grid (Dict[str,float]) – your field in a regular 3D grid.
Returns:: jnp.Array of shape (coords[0].shape). coords is the parameter of your TE_grid instance.