Identity

class pmrf.stats.Identity(_is_constant_jacobian: bool = True, _is_constant_log_det: bool = True)

Bases: AbstractForwardInverseBijector, AbstractInvLogDetJacBijector, AbstractFwdLogDetJacBijector

Identity bijector: y = x.

forward_and_log_det(x: PyTree) → tuple[PyTree, PyTree]

Computes y = x and log|det J(f)(x)| = 0.

inverse_and_log_det(y: PyTree) → tuple[PyTree, PyTree]

Computes x = y and log|det J(f^{-1})(y)| = 0.

same_as(other: AbstractBijector) → bool

Returns True if this bijector is guaranteed to be the same as other.