Radial Flow

Implementation of Radial Flows (RadialFlow) proposed in (https://arxiv.org/abs/1505.05770)

class pythae.models.normalizing_flows.RadialFlowConfig(input_dim=None)[source]

This is the RadialFlow model configuration instance.

Parameters

input_dim (tuple) – The input data dimension. Default: None.

class pythae.models.normalizing_flows.RadialFlow(model_config)[source]

Radial Flow model.

Parameters

model_config (RadialFlowConfig) – The RadialFlow model configuration setting the main parameters of the model.

forward(x, **kwargs)[source]

The input data is transformed toward the prior

Parameters

inputs (torch.Tensor) – An input tensor

Returns

An instance of ModelOutput containing all the relevant parameters

Return type

ModelOutput