Neural Spline¤

Module: generative_models.models.flow.neural_spline

Source: generative_models/models/flow/neural_spline.py

Overview¤

Neural Spline Flows implementation.

Based on the paper: "Neural Spline Flows" by Durkan et al. (2019) https://arxiv.org/abs/1906.04032

This implementation provides rational quadratic spline transformations for normalizing flows, following the existing architecture patterns.

class NeuralSplineFlow

class RationalQuadraticSplineTransform

class SplineCouplingLayer

def __init__()

def __init__()

def __init__()

def apply_spline()

def forward()

def forward()

def generate()

def inverse()

def inverse()

def log_prob()

def loss_fn()

def sample()