If so, would really appreciate it if someone could point me to the definition. Thanks!
How about torch.max?
torch.max(input, dim, keepdim=False, out=None) -> (Tensor, LongTensor)
Returns the maximum value of each row of the input tensor in the given dimension dim.