kd.optim.exclude

kd.optim.exclude#

kauldron.optim.exclude(
pattern: str | collections.abc.Sequence[str],
) collections.abc.Callable[[Any], Any][source]

Create a mask which selects all nodes except the ones matching the pattern.

This is the inverse of select(). All the tree nodes will be True, except the ones matching the pattern.

Parameters:

pattern – The pattern to exclude. See select() for more details.

Returns:

The optax mask factory.