Merge pull request #2625 from parrangoiz/indexkernel-docs-fix

Fix error in IndexKernel docstring
cornellius-gp · Jan 17, 2025 · 4b6d48e · 4b6d48e
2 parents 6bfbd4f + fdbf24a
commit 4b6d48e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gpytorch/kernels/index_kernel.py b/gpytorch/kernels/index_kernel.py
@@ -47,7 +47,8 @@ class IndexKernel(Kernel):
         covar_factor:
             The :math:`B` matrix.
         raw_var:
-            The element-wise log of the :math:`\mathbf v` vector.
+            The element-wise `Softplus <https://pytorch.org/docs/stable/generated/torch.nn.Softplus.html>`_
+            of the :math:`\mathbf v` vector (assuming the default `var_constraint`).
     """
 
     def __init__(