torch.mode¶
- torch.mode(input, dim=-1, keepdim=False, *, out=None)¶
Returns a namedtuple
(values, indices)
wherevalues
is the mode value of each row of theinput
tensor in the given dimensiondim
, i.e. a value which appears most often in that row, andindices
is the index location of each mode value found.By default,
dim
is the last dimension of theinput
tensor.If
keepdim
isTrue
, the output tensors are of the same size asinput
except in the dimensiondim
where they are of size 1. Otherwise,dim
is squeezed (seetorch.squeeze()
), resulting in the output tensors having 1 fewer dimension thaninput
.Note
This function is not defined for
torch.cuda.Tensor
yet.- Parameters
- Keyword Arguments
out (tuple, optional) – the result tuple of two output tensors (values, indices)
Example:
>>> b = torch.tensor( [[0, 0, 0, 2, 0, 0, 2], [0, 3, 0, 0, 2, 0, 1], [2, 2, 2, 0, 0, 0, 3], [2, 2, 3, 0, 1, 1, 0], [1, 1, 0, 0, 2, 0, 2]]) >>> torch.mode(b, 0) torch.return_types.mode( values=tensor([0, 2, 0, 0, 0, 0, 2]), indices=tensor([1, 3, 4, 4, 2, 4, 4]))