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

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

torch.isnan— это функция в PyTorch, используемая для проверки, является ли значение NaN.

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

torch.isnan(input)

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

Пример

import torch
import math

x = torch.tensor([1.0, math.nan, 3.0, float('nan')])

result = torch.isnan(x)

print(result)

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

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

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

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