diff options
Diffstat (limited to 'docs')
| -rw-r--r-- | docs/source/distributions.rst | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/docs/source/distributions.rst b/docs/source/distributions.rst index de541b467e..5a8a00f413 100644 --- a/docs/source/distributions.rst +++ b/docs/source/distributions.rst @@ -257,6 +257,15 @@ Probability distributions - torch.distributions :undoc-members: :show-inheritance: +:hidden:`LogitRelaxedBernoulli` +~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. currentmodule:: torch.distributions.relaxed_bernoulli +.. autoclass:: LogitRelaxedBernoulli + :members: + :undoc-members: + :show-inheritance: + :hidden:`RelaxedOneHotCategorical` ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ |