Функция minimum
Функция minimum вычисляет поэлементный минимум
между двумя тензорами. Первым параметром функция
принимает тензор input, вторым параметром -
тензор other. Функция возвращает новый тензор,
каждый элемент которого равен минимальному значению
из соответствующих элементов входных тензоров.
Тензоры должны быть приводимы к одинаковой форме
с помощью механизма вещания (broadcasting).
Синтаксис
torch.minimum(input, other)
Пример
Давайте вычислим поэлементный минимум двух одномерных тензоров одинаковой длины:
import torch
t1 = torch.tensor([1, 5, 3, 8, 2])
t2 = torch.tensor([4, 2, 7, 1, 6])
res = torch.minimum(t1, t2)
print(res)
Результат выполнения кода:
tensor([1, 2, 3, 1, 2])
Пример
Применим функцию к двумерным тензорам одинаковой формы:
import torch
t1 = torch.tensor([
[10, 20, 30],
[40, 50, 60],
])
t2 = torch.tensor([
[15, 25, 5],
[45, 35, 55],
])
res = torch.minimum(t1, t2)
print(res)
Результат выполнения кода:
tensor([
[10, 20, 5],
[40, 35, 55],
])
Пример
Функция поддерживает механизм вещания, что позволяет сравнивать тензор со скаляром:
import torch
t = torch.tensor([3, 7, 1, 9, 5])
res = torch.minimum(t, 4)
print(res)
Результат выполнения кода:
tensor([3, 4, 1, 4, 4])
Пример
Рассмотрим пример вещания между тензорами разных форм. Сравним матрицу со строкой:
import torch
t1 = torch.tensor([
[1, 2, 3],
[4, 5, 6],
[7, 8, 9],
])
t2 = torch.tensor([4, 5, 6])
res = torch.minimum(t1, t2)
print(res)
Результат выполнения кода:
tensor([
[1, 2, 3],
[4, 5, 6],
[4, 5, 6],
])
