diff --git a/dwave/optimization/mathematical.py b/dwave/optimization/mathematical.py index d326f972..c3c3ea25 100644 --- a/dwave/optimization/mathematical.py +++ b/dwave/optimization/mathematical.py @@ -79,6 +79,7 @@ "absolute", "add", "arange", + "argsort", "as_array_symbols", "atleast_1d", "atleast_2d", @@ -302,16 +303,13 @@ def argsort(array: ArraySymbol) -> ArgSort: >>> a = model.constant([[5, 2, 7], [4, 9, 1]]) >>> indices = argsort(a) >>> indices.shape() - (5,) + (2, 3) >>> with model.lock(): ... model.states.resize(1) ... print(indices.state()) [[5. 1. 3.] [0. 2. 4.]] - See Also: - :class:`~dwave.optimization.ArgSort`: equivalent symbol. - .. versionadded:: 0.6.4 """ return ArgSort(array)