torch.max
Software / App
A PyTorch function that returns the maximum value and its index along a specified dimension of a tensor.
Mentioned in 1 video
A PyTorch function that returns the maximum value and its index along a specified dimension of a tensor.