Add missing docs for ProjectedNormal parameter (#3009)

pyro-ppl · Jan 23, 2022 · 65225ae · 65225ae
1 parent b134ae8
commit 65225ae
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pyro/distributions/projected_normal.py b/pyro/distributions/projected_normal.py
@@ -42,6 +42,10 @@ def model():
         "The General Projected Normal Distribution of Arbitrary Dimension:
         Modeling and Bayesian Inference"
         https://projecteuclid.org/euclid.ba/1453211962
+
+    :param torch.Tensor concentration: A combined location-and-concentration
+        vector. The direction of this vector is the location, and its
+        magnitude is the concentration.
     """
 
     arg_constraints = {"concentration": constraints.real_vector}