pulsatrix::ReparamGrad Struct Reference

The (grad_mu, grad_log_sigma) pair both VAE building blocks produce. More...

#include <reparameterize.hpp>

Collaboration diagram for pulsatrix::ReparamGrad:

Public Attributes
Tensor	grad_mu
Tensor	grad_log_sigma

Detailed Description

The (grad_mu, grad_log_sigma) pair both VAE building blocks produce.

Note

Shared by Reparameterize::backward() and KLDivergenceLoss::backward() – both differentiate w.r.t. exactly the same two tensors, so duplicating the struct would mean a caller combining reconstruction and KL gradients (the whole point of the VAE ELBO) juggling two structurally identical but unrelated types. Declared here rather than in a new "VAE types" header: one pair struct does not justify a header of its own, and mission_vae_module.md explicitly rules out a speculative larger shared header beyond this type.

Member Data Documentation

grad_log_sigma

Tensor pulsatrix::ReparamGrad::grad_log_sigma

grad_mu

Tensor pulsatrix::ReparamGrad::grad_mu

---

The documentation for this struct was generated from the following file:

• include/pulsatrix/reparameterize.hpp