Minor documentation mistake (#15068)
Summary:
keepdim is a optional parameter for torch.max()
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15068
Differential Revision: D13437745
Pulled By: zou3519
fbshipit-source-id: b5198c7d4ae17758cd136f6e5aecc6cb5838f174
