Функция PyTorch torch.isinf
Справочное руководство Pytorch torch
torch.isinfявляется функцией в PyTorch, используемой для проверки того, является ли значение бесконечным. Возвращает булев тензор, в котором для каждого элемента значение True, если значение бесконечное (положительная бесконечность или отрицательная бесконечность).
Определение функции
torch.isinf(input)
Пример использования
Пример
import torch
# Создать тензор, содержащий разные значения
x = torch.tensor([1.0, float('inf'), float('-inf'), 2.0])
# Проверить, являются ли значения бесконечными
result = torch.isinf(x)
print(result)
# Создать тензор, содержащий разные значения
x = torch.tensor([1.0, float('inf'), float('-inf'), 2.0])
# Проверить, являются ли значения бесконечными
result = torch.isinf(x)
print(result)
Результат вывода:
tensor([False, True, True, False])
Другие расширения