autopdex.assembler.user_residual_assemble_tangent_diagonal

autopdex.assembler.user_residual_assemble_tangent_diagonal(dofs, settings, static_settings, set)

Assembly of the diagonal of the tangent matrix for custom user residual of specified domain.

Parameters:

• dofs (jnp.ndarray or dict) – Degrees of freedom.

• settings (dict) – Settings dictionary.

• static_settings (flax.core.FrozenDict) – Static settings as frozen dictionary.

• set (int) – The domain number.

Returns:

The diagonal of the assembled tangent matrix.

Return type:

jnp.ndarray