autopdex.assembler.user_element_assemble_tangent

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

Assembly of the full (sparse) tangent matrix for custom user element 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 assembled tangent matrix.

Return type:

jax.experimental.sparse.BCOO