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