diff --git a/gpytorch/kernels/index_kernel.py b/gpytorch/kernels/index_kernel.py index 7fa5e01f3..cd060ab6b 100644 --- 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 `_ + of the :math:`\mathbf v` vector (assuming the default `var_constraint`). """ def __init__(