autopdex.models.heat_equation

autopdex.models.heat_equation(diffusivity_fun)

Space-time heat equation with thermal diffusivity alpha.

Last dimension is the time dimension, remaining dimensions are spatial dimensions

Parameters:

diffusivity_fun (function) – Function to compute the thermal diffusivity.

Returns:

Function to compute the PDE residual.

Return type:

heat_fun (function)

Notes

• This models works for DOFs as a jnp.ndarray.