torch.mode

torch.mode(input, dim=-1, keepdim=False, *, out=None)

Возвращает кортеж с именем (values, indices) , где values — значение моды каждой строки тензора input в заданном измерении dim, т.е. значение, которое встречается чаще всего в этой строке, а indices — индекс местоположения каждого значения моды.

По умолчанию, dim является последним измерением тензора input.

Если keepdim равно True, выходные тензоры имеют такой же размер, как input, за исключением измерения dim, где они имеют размер 1. В противном случае, dim сжимается (см. torch.squeeze()), в результате чего выходные тензоры имеют на одно измерение меньше, чем input.

Примечание

Эта функция пока не определена для torch.cuda.Tensor.

Параметры

• input (Тензор) – входной тензор.
• dim (int) – измерение для сокращения.
• keepdim (bool) – сохранять ли размерность выхода.

Ключевые аргументы

out (кортеж, необязательно) – кортеж из двух выходных тензоров (значения, индексы)

Пример:

>>> a = torch.randint(10, (5,))
>>> a
tensor([6, 5, 1, 0, 2])
>>> b = a + (torch.randn(50, 1) * 5).long()
>>> torch.mode(b, 0)
torch.return_types.mode(values=tensor([6, 5, 1, 0, 2]), indices=tensor([2, 2, 2, 2, 2]))