Tied

class pmrf.modules.Tied(module: ~pmrf.modules.base.Module | None = None, target: ~typing.Callable[[~typing.Any], ~typing.Any] | None = None, source: ~typing.Callable[[~typing.Any], ~typing.Any] | None = None, tie_fn: ~typing.Callable[[~typing.Any], ~typing.Any] = <function Tied.<lambda>>, *, model: ~pmrf.modules.base.Module | None = None)

Bases: Module, AbstractUnwrappable

A module that ties a target field to a transformed source field.

named_params(full_params: bool = False, free_only: bool = False, namespace_separator: str = '_') → dict[str, float | Array | Param]

Return names relative to the wrapped module rather than the wrapper.

unwrap() → Module

Evaluates and returns the underlying unwrapped PyTree node.

Returns:

The resolved underlying value, assuming no wrapped subnodes exist.

property module: Module

The underlying module with tied targets removed.

tie: Tie