VAE reparameterization trick: z = mu + exp(log_sigma) * epsilon. More...

#include "pulsatrix/device_backend.hpp"
#include "pulsatrix/tensor.hpp"

Include dependency graph for reparameterize.hpp:

This graph shows which files directly or indirectly include this file:

Classes
struct	pulsatrix::ReparamGrad
	The (grad_mu, grad_log_sigma) pair both VAE building blocks produce. More...

class	pulsatrix::Reparameterize
	VAE reparameterization trick (Kingma & Welling 2013, arXiv:1312.6114), z[b,d] = mu[b,d] + exp(log_sigma[b,d]) * epsilon[b,d]. More...

Detailed Description

VAE reparameterization trick: z = mu + exp(log_sigma) * epsilon.

Classes