bijx.RayTransform

class bijx.RayTransform

Bases: Bijection

Simple ray transformation ensuring f(0)=0.

__init__(bijection)

Parameters:

bijection (Bijection)

Methods

forward(x, ld)	Apply forward transformation.
invert()	Create an inverted version of this bijection.
reverse(y, ld)	Apply reverse (inverse) transformation.

forward(x, ld)

Apply forward transformation.

Transforms input through the bijection and updates log-density according to the change of variables formula.

For convenience Bijection() gives the default identity bijection.

Parameters:

• x – Input data of any pytree structure.

• log_density – Log density values corresponding to the input.

• **kwargs – Additional transformation-specific arguments.

Returns:

Tuple of (transformed_data, updated_log_density) where the log-density incorporates the log absolute determinant of the transformation Jacobian.

reverse(y, ld)

Apply reverse (inverse) transformation.

Transforms input through the inverse bijection and updates log-density accordingly.

Parameters:

• x – Input data of any pytree structure.

• log_density – Log density values corresponding to the input.

• **kwargs – Additional transformation-specific arguments.

Returns:

Tuple of (inverse_transformed_data, updated_log_density) where the log-density change has the opposite sign compared to forward().