autopdex.assembler.assemble_tangent_diagonal

autopdex.assembler.assemble_tangent_diagonal(dofs, settings, static_settings)[source]

Assemble the diagonal of the tangent matrix.

Parameters:

dofs (jnp.ndarray or dict) – Degrees of freedom.
settings (dict) – Settings dictionary.
static_settings (flax.core.FrozenDict) – Static settings as frozen dictionary.

Returns:

The diagonal of the assembled tangent matrix.

Return type:

jnp.ndarray