invertible_sequential module

class InvertibleSequentialFlow(*args, **kwargs)[source]

Bases: zunis.models.flows.general_flow.GeneralFlow

Initializes internal Module state, shared by both nn.Module and ScriptModule.

flow(x)[source]

Transform the batch of points x with shape (…,d) This is an abstract method that should be overriden

invert()[source]
training: bool[source]
transform_and_compute_jacobian(xj)[source]

Compute the flow transformation and its Jacobian simulatenously on xj with xj.shape == (…,d+1)

This is an abstract method that should be overriden