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) – сохраняется ли dim в выходном тензоре или нет.

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

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]))