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