BijectorCalculator#

class liesel.liesel.nodes.BijectorCalculator(bijector, input, inverse=False)[source]#

Bases: liesel.liesel.nodes.NodeCalculator

Evaluates the forward or the inverse method of a TFP bijector at its input.

Methods

jaxify()

Enables JAX NumPy for the node component.

unjaxify()

Disables JAX NumPy for the node component.

value()

Calculates the value of a node from its inputs and returns it.

Attributes

bijector

rtype

Any

inputs

The inputs of the node component.

jaxified

Whether JAX NumPy is enabled for the node component.