Функция torch.minimum в PyTorch
Справочное руководство Pytorch torch
torch.minimumЭто функция в PyTorch, используемая для поэлементного взятия минимального значения. Она возвращает новый тензор, составленный из минимальных значений соответствующих элементов двух тензоров.
Определение функции
torch.minimum(input, other)
Пример использования
Пример
import torch
# Создайте два тензора
x = torch.tensor([3, 1, 4])
y = torch.tensor([1, 5, 2])
# Поэлементно взять минимум
result = torch.minimum(x, y)
print(result)
# Создайте два тензора
x = torch.tensor([3, 1, 4])
y = torch.tensor([1, 5, 2])
# Поэлементно взять минимум
result = torch.minimum(x, y)
print(result)
Результат вывода:
tensor([1, 1, 2])
Другие расширения