Method sample

roach/models/distributions.py:251–253

(self)

        return th.mean(exploration_loss)

    def sample(self) -> th.Tensor:
        # Reparametrization trick to pass gradients
        return self.distribution.rsample()

    def mode(self) -> th.Tensor:
        alpha = self.distribution.concentration1