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.