fix typo in argmax docstring (pytorch#55239)

Summary:
argmax docstring previously said that it returns indexes of the first 'minimal' value, fixed typo in that line to 'maximal'

Pull Request resolved: pytorch#55239

Reviewed By: albanD

Differential Revision: D27641562

Pulled By: mrshenli

fbshipit-source-id: f8b5c579400088b5210c83a05da6c4c106fbf95d

torch/_torch_docs.py

@@ -5206,7 +5206,7 @@ def merge_dicts(*dicts):
 This is the second value returned by :meth:`torch.max`. See its
 documentation for the exact semantics of this method.
 
-.. note:: If there are multiple minimal values then the indices of the first minimal value are returned.
+.. note:: If there are multiple maximal values then the indices of the first maximal value are returned.
 
 Args:
     {input}