Method sample

roach/models/distributions.py:162–163

(self)

        return None

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

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