Pytorch equivalent of TF reduce_max

Is there a reduce_max equivalent of TF(can take multiple dims as input)?
the torch.max operation allows only one dim as input.
I require this for implementation of RMAC. If there are any other alternatives please suggest.

1 Like

Is this what you are looking for ?