SYMBOL INDEX (94 symbols across 11 files)

FILE: main_experiment.py
  function run (line 106) | def run(args, kwargs):

FILE: models/VAE.py
  class VAE (line 10) | class VAE(nn.Module):
    method __init__ (line 16) | def __init__(self, args):
    method create_encoder (line 45) | def create_encoder(self):
    method create_decoder (line 88) | def create_decoder(self):
    method reparameterize (line 133) | def reparameterize(self, mu, var):
    method encode (line 146) | def encode(self, x):
    method decode (line 158) | def decode(self, z):
    method forward (line 170) | def forward(self, x):
  class PlanarVAE (line 185) | class PlanarVAE(VAE):
    method __init__ (line 190) | def __init__(self, args):
    method encode (line 210) | def encode(self, x):
    method forward (line 229) | def forward(self, x):
  class OrthogonalSylvesterVAE (line 254) | class OrthogonalSylvesterVAE(VAE):
    method __init__ (line 259) | def __init__(self, args):
    method batch_construct_orthogonal (line 318) | def batch_construct_orthogonal(self, q):
    method encode (line 362) | def encode(self, x):
    method forward (line 398) | def forward(self, x):
  class HouseholderSylvesterVAE (line 428) | class HouseholderSylvesterVAE(VAE):
    method __init__ (line 433) | def __init__(self, args):
    method batch_construct_orthogonal (line 486) | def batch_construct_orthogonal(self, q):
    method encode (line 516) | def encode(self, x):
    method forward (line 552) | def forward(self, x):
  class TriangularSylvesterVAE (line 585) | class TriangularSylvesterVAE(VAE):
    method __init__ (line 591) | def __init__(self, args):
    method encode (line 637) | def encode(self, x):
    method forward (line 671) | def forward(self, x):
  class IAFVAE (line 704) | class IAFVAE(VAE):
    method __init__ (line 709) | def __init__(self, args):
    method encode (line 723) | def encode(self, x):
    method forward (line 736) | def forward(self, x):

FILE: models/flows.py
  class Planar (line 15) | class Planar(nn.Module):
    method __init__ (line 21) | def __init__(self):
    method der_h (line 28) | def der_h(self, x):
    method forward (line 33) | def forward(self, zk, u, w, b):
  class Sylvester (line 68) | class Sylvester(nn.Module):
    method __init__ (line 73) | def __init__(self, num_ortho_vecs):
    method der_h (line 88) | def der_h(self, x):
    method der_tanh (line 91) | def der_tanh(self, x):
    method _forward (line 94) | def _forward(self, zk, r1, r2, q_ortho, b, sum_ldj=True):
    method forward (line 140) | def forward(self, zk, r1, r2, q_ortho, b, sum_ldj=True):
  class TriangularSylvester (line 145) | class TriangularSylvester(nn.Module):
    method __init__ (line 150) | def __init__(self, z_size):
    method der_h (line 160) | def der_h(self, x):
    method der_tanh (line 163) | def der_tanh(self, x):
    method _forward (line 166) | def _forward(self, zk, r1, r2, b, permute_z=None, sum_ldj=True):
    method forward (line 219) | def forward(self, zk, r1, r2, q_ortho, b, sum_ldj=True):
  class IAF (line 224) | class IAF(nn.Module):
    method __init__ (line 239) | def __init__(self, z_size, num_flows=2, num_hidden=0, h_size=50, forge...
    method forward (line 284) | def forward(self, z, h_context):

FILE: models/layers.py
  class Identity (line 8) | class Identity(nn.Module):
    method __init__ (line 9) | def __init__(self):
    method forward (line 12) | def forward(self, x):
  class GatedConv2d (line 16) | class GatedConv2d(nn.Module):
    method __init__ (line 17) | def __init__(self, input_channels, output_channels, kernel_size, strid...
    method forward (line 26) | def forward(self, x):
  class GatedConvTranspose2d (line 37) | class GatedConvTranspose2d(nn.Module):
    method __init__ (line 38) | def __init__(self, input_channels, output_channels, kernel_size, strid...
    method forward (line 50) | def forward(self, x):
  class MaskedLinear (line 61) | class MaskedLinear(nn.Module):
    method __init__ (line 70) | def __init__(self, in_features, out_features, diagonal_zeros=False, bi...
    method reset_parameters (line 86) | def reset_parameters(self):
    method build_mask (line 91) | def build_mask(self):
    method forward (line 110) | def forward(self, x):
    method __repr__ (line 118) | def __repr__(self):
  class MaskedConv2d (line 130) | class MaskedConv2d(nn.Module):
    method __init__ (line 139) | def __init__(self, in_features, out_features, size_kernel=(3, 3), diag...
    method reset_parameters (line 156) | def reset_parameters(self):
    method build_mask (line 161) | def build_mask(self):
    method forward (line 188) | def forward(self, x):
    method __repr__ (line 192) | def __repr__(self):

FILE: optimization/loss.py
  function binary_loss_function (line 11) | def binary_loss_function(recon_x, x, z_mu, z_var, z_0, z_k, ldj, beta=1.):
  function multinomial_loss_function (line 53) | def multinomial_loss_function(x_logit, x, z_mu, z_var, z_0, z_k, ldj, ar...
  function binary_loss_array (line 101) | def binary_loss_array(recon_x, x, z_mu, z_var, z_0, z_k, ldj, beta=1.):
  function multinomial_loss_array (line 127) | def multinomial_loss_array(x_logit, x, z_mu, z_var, z_0, z_k, ldj, args,...
  function cross_entropy (line 160) | def cross_entropy(input, target, weight=None, size_average=True, ignore_...
  function nll_loss (line 188) | def nll_loss(input, target, weight=None, size_average=True, ignore_index...
  function calculate_loss (line 234) | def calculate_loss(x_mean, x, z_mu, z_var, z_0, z_k, ldj, args, beta=1.):
  function calculate_loss_array (line 253) | def calculate_loss_array(x_mean, x, z_mu, z_var, z_0, z_k, ldj, args):

FILE: optimization/training.py
  function train (line 12) | def train(epoch, train_loader, model, opt, args):
  function evaluate (line 71) | def evaluate(data_loader, model, args, testing=False, file=None, epoch=0):

FILE: utils/distributions.py
  function log_normal_diag (line 20) | def log_normal_diag(x, mean, log_var, average=False, reduce=True, dim=No...
  function log_normal_normalized (line 31) | def log_normal_normalized(x, mean, log_var, average=False, reduce=True, ...
  function log_normal_standard (line 46) | def log_normal_standard(x, average=False, reduce=True, dim=None):
  function log_bernoulli (line 58) | def log_bernoulli(x, mean, average=False, reduce=True, dim=None):

FILE: utils/load_data.py
  function load_static_mnist (line 13) | def load_static_mnist(args, **kwargs):
  function load_freyfaces (line 57) | def load_freyfaces(args, **kwargs):
  function load_omniglot (line 104) | def load_omniglot(args, **kwargs):
  function load_caltech101silhouettes (line 156) | def load_caltech101silhouettes(args, **kwargs):
  function load_dataset (line 192) | def load_dataset(args, **kwargs):

FILE: utils/log_likelihood.py
  function calculate_likelihood (line 7) | def calculate_likelihood(X, model, args, S=5000, MB=500):

FILE: utils/plotting.py
  function plot_training_curve (line 11) | def plot_training_curve(train_loss, validation_loss, fname='training_cur...

FILE: utils/visual_evaluation.py
  function plot_reconstructions (line 9) | def plot_reconstructions(data, recon_mean, loss, loss_type, epoch, args):
  function plot_images (line 32) | def plot_images(args, x_sample, dir, file_name, size_x=3, size_y=3):