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