Merge pull request #19445 from timhoffm/doc-broadcast_to

DOC: broadcast_to() supports int as shape parameter
numpy · Jul 12, 2021 · c07080f · c07080f
2 parents bb71b5f + 2599a4c
commit c07080f
Showing 1 changed file with 3 additions and 2 deletions.
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).