diff --git a/torch/_torch_docs.py b/torch/_torch_docs.py index d225b030e27e..fb92f892b6e1 100644 --- a/torch/_torch_docs.py +++ b/torch/_torch_docs.py @@ -2447,11 +2447,9 @@ def merge_dicts(*dicts): out[i][j][k] = input[i][index[i][j][k]][k] # if dim == 1 out[i][j][k] = input[i][j][index[i][j][k]] # if dim == 2 -If :attr:`input` is an n-dimensional tensor with size -:math:`(x_0, x_1..., x_{i-1}, x_i, x_{i+1}, ..., x_{n-1})` -and ``dim = i``, then :attr:`index` must be an :math:`n`-dimensional tensor with -size :math:`(x_0, x_1, ..., x_{i-1}, y, x_{i+1}, ..., x_{n-1})` where :math:`y \geq 1` -and :attr:`out` will have the same size as :attr:`index`. +:attr:`input` and :attr:`index` must have the same number of dimensions. +It is also required that ``index.size(d) <= input.size(d)`` for all +dimensions ``d != dim``. :attr:`out` will have the same shape as :attr:`index`. """ + r""" Args: input (Tensor): the source tensor