Transformed

class pmrf.constraints.Transformed(constraint: AbstractConstraint, bijector: AbstractBijector)

Bases: AbstractConstraint

A constraint modified by an arbitrary distreqx bijector.

The custom bijector is applied after the base constraint. This allows for complex normalizations or transformations on top of physical boundaries.

Attributes:

base_constraint: The underlying physical constraint applied first. bijector: The bijector applied on top of the base constraint.

property base_bijector: AbstractBijector
property base_bounds: tuple[PyTree, PyTree]
base_constraint: AbstractConstraint
bijector: AbstractBijector
bounds: tuple[PyTree, PyTree]
transform_bijector: AbstractBijector