Hi,
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.