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