[Docs] Fix typo in torch.unflatten (pytorch#109588)

Fixes pytorch#109559 Pull Request resolved: pytorch#109588 Approved by: https://github.com/lezcano
heysaeed · Sep 19, 2023 · 2f53bca · 2f53bca
1 parent af867c2
commit 2f53bca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch/_torch_docs.py b/torch/_torch_docs.py
@@ -4555,7 +4555,7 @@ def merge_dicts(*dicts):
     torch.Size([3, 2, 2, 1])
     >>> torch.unflatten(torch.randn(3, 4, 1), 1, (-1, 2)).shape
     torch.Size([3, 2, 2, 1])
-    >>> torch.unflatten(torch.randn(5, 12, 3), -1, (2, 2, 3, 1, 1)).shape
+    >>> torch.unflatten(torch.randn(5, 12, 3), -2, (2, 2, 3, 1, 1)).shape
     torch.Size([5, 2, 2, 3, 1, 1, 3])
 """.format(
         **common_args