dimensions to reduce.
  * **keepdim** (*bool*) -- whether the output tensor has "dim"
    retained or not.

Keyword Arguments:
      out (Tensor, optional) -- the output tensor.
Example:
  >>> a = torch.randn(4, 4)
  >>> a
  tensor([[ 0.8177,  1.4878, -0.2491,  0.9130],
          [-0.7158,  1.1775,  2.0992,  0.4817],
          [-0.0053,  0.0164, -1.3738, -0.0507],
          [ 1.9700,  1.1106, -1.0318, -1.0816]])
  >>> torch.amax(a, 1)
  tensor([1.4878, 2.0992, 0.0164, 1.9700])
