Method sample

roach/models/distributions.py:100–101

(self)

        return th.mean(exploration_loss)

    def sample(self) -> th.Tensor:
        return self.distribution.rsample()

    def mode(self) -> th.Tensor:
        return self.distribution.mean