Функция torch.fmax в PyTorch

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.fmaxЭто функция в PyTorch для поэлементного максимума (с игнорированием NaN). Когда входные данные содержат NaN, torch.maximum возвращает NaN, а torch.fmax игнорирует NaN и возвращает значения, не равные NaN.

Определение функции

torch.fmax(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.fmax(x, y)
print(result)

Результат вывода:

tensor([3., 5., 4.])

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения