Функция torch.median в PyTorch
Справочное руководство PyTorch torch
torch.medianЭто функция в PyTorch, используемая для возврата медианы тензора. Медиана — это значение в середине после сортировки данных.
Определение функции
torch.median(input, dim, keepdim=False)
Пример использования
Пример
import torch
x = torch.tensor([4, 2, 1, 3, 5])
# Возвращает медиану всех элементов
print("Глобальная медиана:", torch.median(x))
# Медиана вдоль dim=0
y = torch.tensor([[1, 3, 2], [4, 1, 3]])
print("Медиана по dim=0:", torch.median(y, dim=0))
x = torch.tensor([4, 2, 1, 3, 5])
# Возвращает медиану всех элементов
print("Глобальная медиана:", torch.median(x))
# Медиана вдоль dim=0
y = torch.tensor([[1, 3, 2], [4, 1, 3]])
print("Медиана по dim=0:", torch.median(y, dim=0))
Результат вывода:
全局中位数: tensor(3) dim=0 中位数: torch.return_types.median(values=tensor([1, 1, 2]), indices=tensor([0, 1, 0]))
Другие расширения