archimedes.tree.is_leaf

archimedes.tree.is_leaf(x: Any) → bool

Check if a value is a leaf in a pytree.

Returns True if the value is not a container (i.e. is an array, a scalar, or None).

Parameters:

x (Any) – The value to check.

Returns:

True if the value is a leaf, False otherwise.

Return type:

bool