REINFORCE's return-weighted negative log-likelihood loss over a batched rollout. More...

#include <cstdint>
#include <vector>
#include "pulsatrix/device_backend.hpp"
#include "pulsatrix/tensor.hpp"

Include dependency graph for policy_gradient_loss.hpp:

Classes
class	pulsatrix::PolicyGradientLoss
	loss = mean_b( -log pi(a_b \| s_b) * G_b ), where `a_b` is the action actually taken on step `b` of a rollout and `G_b` its return – the REINFORCE policy-gradient surrogate of Williams 1992. More...

Detailed Description

REINFORCE's return-weighted negative log-likelihood loss over a batched rollout.

Classes