We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 975dba9 commit cea9e6dCopy full SHA for cea9e6d
gpytorch/variational/lmc_variational_strategy.py
@@ -178,7 +178,7 @@ def __call__(
178
The return type will be a (... x N)
179
:class:`~gpytorch.distributions.MultivariateNormal`.
180
181
- :param x: Input locations to evaluate variational strategy
+ :param x: (... x N x D) Input locations to evaluate variational strategy
182
:param task_indices: (Default: None) Task index associated with each input.
183
If this **is not** provided, then the returned distribution evaluates every input on every task
184
(returns :class:`~gpytorch.distributions.MultitaskMultivariateNormal`).
0 commit comments