Функция torch.mode в PyTorch

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.modeЭто функция в PyTorch, используемая для возврата моды тензора. Мода — это значение, которое чаще всего встречается в наборе данных.

Определение функции

torch.mode(input, dim, keepdim=False)

Пример использования

Пример

import torch

x = torch.tensor([1, 2, 2, 3, 3, 3])

# Возвращаем моду
print("Мода:", torch.mode(x))

# Мода по dim=0
y = torch.tensor([[1, 2, 2], [1, 2, 3]])
print("мода по dim=0:", torch.mode(y, dim=0))

# Мода по dim=1
print("мода по dim=1:", torch.mode(y, dim=1))

Результат вывода:

众数: torch.return_types.mode(values=tensor(3), indices=tensor(4))
dim=0 众数: torch.return_types.mode(values=tensor([1, 2, 2]), indices=tensor([0, 0, 0]))
dim=1 众数: torch.return_types.mode(values=tensor([2, 3]), indices=tensor([1, 2]))

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения