element_wise module¶

class AnalyticInverseSigmoid(*args, **kwargs)[source]¶

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

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

class AnalyticSigmoid(*args, **kwargs)[source]¶

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

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

class InvertibleAnalyticSigmoid(*args, **kwargs)[source]¶

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

ZüNIS