Fix the basic description of torch.min(), torch.max(), torch.all(), torch.any() (#152658)

ILCSFNO · pytorchmergebot · commit 7cb5c751c3f7 · 2025-05-08T22:59:14.000Z
Fixes #152176 Pull Request resolved: #152658 Approved by: https://github.com/malfet
diff --git a/torch/_torch_docs.py b/torch/_torch_docs.py
@@ -792,14 +792,20 @@ def merge_dicts(*dicts):
 add_docstr(
     torch.all,
     r"""
-all(input: Tensor) -> Tensor
+all(input: Tensor, *, out=None) -> Tensor
 
 Tests if all elements in :attr:`input` evaluate to `True`.
 
 .. note:: This function matches the behaviour of NumPy in returning
           output of dtype `bool` for all supported dtypes except `uint8`.
           For `uint8` the dtype of output is `uint8` itself.
 
+Args:
+    {input}
+
+Keyword args:
+    {out}
+
 Example::
 
     >>> a = torch.rand(1, 2).bool()
@@ -855,6 +861,12 @@ def merge_dicts(*dicts):
           output of dtype `bool` for all supported dtypes except `uint8`.
           For `uint8` the dtype of output is `uint8` itself.
 
+Args:
+    {input}
+
+Keyword args:
+    {out}
+
 Example::
 
     >>> a = torch.rand(1, 2).bool()
@@ -6488,13 +6500,16 @@ def merge_dicts(*dicts):
 add_docstr(
     torch.max,
     r"""
-max(input) -> Tensor
+max(input, *, out=None) -> Tensor
 
 Returns the maximum value of all elements in the ``input`` tensor.
 
 Args:
     {input}
 
+Keyword args:
+    {out}
+
 Example::
 
     >>> a = torch.randn(1, 3)
@@ -7103,13 +7118,16 @@ def merge_dicts(*dicts):
 add_docstr(
     torch.min,
     r"""
-min(input) -> Tensor
+min(input, *, out=None) -> Tensor
 
 Returns the minimum value of all elements in the :attr:`input` tensor.
 
 Args:
     {input}
 
+Keyword args:
+    {out}
+
 Example::
 
     >>> a = torch.randn(1, 3)
