Fonction torch.isnan de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.isnanest une fonction de PyTorch utilisée pour vérifier si une valeur est NaN.

Définition de fonction

torch.isnan(input)

Exemple d'utilisation

Exemple

import torch
import math

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

result = torch.isnan(x)

print(result)

Le résultat de sortie est:

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

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions