Skip to content

Commit cea9e6d

Browse files
committed
Include side in LMCVarStrat docstring
1 parent 975dba9 commit cea9e6d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

gpytorch/variational/lmc_variational_strategy.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,7 @@ def __call__(
178178
The return type will be a (... x N)
179179
:class:`~gpytorch.distributions.MultivariateNormal`.
180180
181-
:param x: Input locations to evaluate variational strategy
181+
:param x: (... x N x D) Input locations to evaluate variational strategy
182182
:param task_indices: (Default: None) Task index associated with each input.
183183
If this **is not** provided, then the returned distribution evaluates every input on every task
184184
(returns :class:`~gpytorch.distributions.MultitaskMultivariateNormal`).

0 commit comments

Comments
 (0)