Iaf

Module: generative_models.models.flow.iaf

Source: generative_models/models/flow/iaf.py

Overview

Inverse Autoregressive Flow (IAF) implementation.

Based on the paper "Improved Variational Inference with Inverse Autoregressive Flow" by Kingma et al. Reference implementations from benchmark_VAE and other sources are used.

Classes

IAF

class IAF

IAFLayer

class IAFLayer

Functions

call

def __call__()

init

def __init__()

init

def __init__()

forward

def forward()

forward

def forward()

inverse

def inverse()

inverse

def inverse()

log_prob

def log_prob()

sample

def sample()

Module Statistics

• Classes: 2
• Functions: 9
• Imports: 7