analytic_flow module

class AnalyticFlow(*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

training: bool[source]
class InvertibleAnalyticFlow(*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]
runs_forward()[source]

Check the running mode: True if forward and False if backward/inverse

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