Функция torch.fmin в PyTorch
Справочное руководство Pytorch torch
torch.fminЭто функция в PyTorch для поэлементного минимума (с игнорированием NaN). Когда входные данные содержат NaN, torch.minimum возвращает NaN, а torch.fmin игнорирует NaN и возвращает значение, не равное NaN.
Определение функции
torch.fmin(input, other)
Пример использования
Пример
import torch
# Создать тензор, содержащий NaN
x = torch.tensor([3.0, float('nan'), 4.0])
y = torch.tensor([1.0, 5.0, 2.0])
# Поэлементный минимум (с игнорированием NaN)
result = torch.fmin(x, y)
print(result)
# Создать тензор, содержащий NaN
x = torch.tensor([3.0, float('nan'), 4.0])
y = torch.tensor([1.0, 5.0, 2.0])
# Поэлементный минимум (с игнорированием NaN)
result = torch.fmin(x, y)
print(result)
Результат вывода:
tensor([1., 5., 2.])
Другие расширения