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.

Параметры:

• input (Tensor) – входной тензор.
• dim (int, необязательно) – измерение, по которому выполняется сведение.
• keepdim (bool, необязательно) – следует ли сохранять dim выходного тензора. По умолчанию: False.

Именованные аргументы:

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

Пример:

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