autopdex.geometry.subelem_in_elems

autopdex.geometry.subelem_in_elems(subelem_to_find, in_elems)

Find the element that contains a given sub-element. Returns only one element (Only to be used for these cases).

Parameters:

• subelem_to_find (jnp.ndarray) – The sub-element to be found.

• in_elems (jnp.ndarray) – Array of elements to be checked.

Returns:

The element that contains the sub-element.

Return type:

jnp.ndarray