Expose documentation for LKJCholesky distribution (#52904)

This is already added to the master branch in #52763.
pytorch · Feb 26, 2021 · 11c78e9 · 11c78e9
1 parent d6943ea
commit 11c78e9
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/docs/source/distributions.rst b/docs/source/distributions.rst
@@ -176,6 +176,15 @@ Probability distributions - torch.distributions
     :undoc-members:
     :show-inheritance:
 
+:hidden:`LKJCholesky`
+~~~~~~~~~~~~~~~~~~~~~~~
+
+.. currentmodule:: torch.distributions.lkj_cholesky
+.. autoclass:: LKJCholesky
+    :members:
+    :undoc-members:
+    :show-inheritance:
+
 :hidden:`Laplace`
 ~~~~~~~~~~~~~~~~~~~~~~~
 
@@ -337,7 +346,7 @@ Probability distributions - torch.distributions
     :members:
     :undoc-members:
     :show-inheritance:
-    
+
 :hidden:`Weibull`
 ~~~~~~~~~~~~~~~~~~~~~~~