Функция torch.cummin в PyTorch
Справочное руководство Pytorch torch
torch.cummin— это функция в PyTorch, используемая для вычисления кумулятивного минимума. Она возвращает кумулятивный минимум вдоль указанного измерения вместе с соответствующими индексами.
Определение функции
torch.cummin(input, dim, dtype=None)
Примеры использования
Пример
import torch
# Вычислить кумулятивный минимум
x = torch.tensor([3, 1, 4, 1, 5, 9, 2, 6])
values, indices = torch.cummin(x, dim=0)
print("Вход:", x)
print("Кумулятивный минимум:", values)
print("Индексы минимумов:", indices)
# 2D тензор
x = torch.tensor([[3, 1, 4], [1, 5, 9], [2, 6, 5]], dtype=torch.float32)
# Накопление по столбцам
values_col, indices_col = torch.cummin(x, dim=0)
print("nНакопление по столбцам:")
print("Кумулятивный минимум:")
print(values_col)
print("Индексы:")
print(indices_col)
# Накопление по строкам
values_row, indices_row = torch.cummin(x, dim=1)
print("nНакопление по строкам:")
print("Кумулятивный минимум:")
print(values_row)
# Вычислить кумулятивный минимум
x = torch.tensor([3, 1, 4, 1, 5, 9, 2, 6])
values, indices = torch.cummin(x, dim=0)
print("Вход:", x)
print("Кумулятивный минимум:", values)
print("Индексы минимумов:", indices)
# 2D тензор
x = torch.tensor([[3, 1, 4], [1, 5, 9], [2, 6, 5]], dtype=torch.float32)
# Накопление по столбцам
values_col, indices_col = torch.cummin(x, dim=0)
print("nНакопление по столбцам:")
print("Кумулятивный минимум:")
print(values_col)
print("Индексы:")
print(indices_col)
# Накопление по строкам
values_row, indices_row = torch.cummin(x, dim=1)
print("nНакопление по строкам:")
print("Кумулятивный минимум:")
print(values_row)
Другие расширения