Функция PyTorch torch.isfinite

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

torch.isfiniteЭто функция в PyTorch, используемая для проверки, являются ли элементы тензора конечными значениями. Она возвращает булев тензор, где для каждого элемента значение True, если значение конечно (не бесконечность и не NaN).

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

torch.isfinite(input)

Пример использования

Пример

import torch

# Создать тензор, содержащий разные значения
x = torch.tensor([1.0, float('inf'), float('nan'), 2.0])

# Проверить конечные значения
result = torch.isfinite(x)
print(result)

Вывод:

tensor([True, False, False, True])

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

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