pulsatrix::KLDivergenceLoss Member List

This is the complete list of members for pulsatrix::KLDivergenceLoss, including all inherited members.

backward() const	pulsatrix::KLDivergenceLoss
forward(const Tensor &mu, const Tensor &log_sigma)	pulsatrix::KLDivergenceLoss
KLDivergenceLoss(DeviceBackend *backend)	pulsatrix::KLDivergenceLoss	explicit