Merge pull request #19445 from timhoffm/doc-broadcast_to

mattip · web-flow · commit c07080f12652 · 2021-07-12T13:03:28.000+03:00
DOC: broadcast_to() supports int as shape parameter
diff --git a/numpy/lib/stride_tricks.py b/numpy/lib/stride_tricks.py
@@ -371,8 +371,9 @@ def broadcast_to(array, shape, subok=False):
     ----------
     array : array_like
         The array to broadcast.
-    shape : tuple
-        The shape of the desired array.
+    shape : tuple or int
+        The shape of the desired array. A single integer ``i`` is interpreted
+        as ``(i,)``.
     subok : bool, optional
         If True, then sub-classes will be passed-through, otherwise
         the returned array will be forced to be a base-class array (default).
