Fonction torch.is_floating_point de PyTorch
Manuel de référence PyTorch torch
torch.is_floating_pointIl s'agit d'une fonction de PyTorch utilisée pour vérifier si le type de données d'un tenseur est un type à virgule flottante.
Définition de fonction
torch.is_floating_point(input)
Exemple d'utilisation
Exemple
import torch
# Tenseur à virgule flottante
x = torch.tensor([1.0, 2.0, 3.0])
print("Virgule flottante :", torch.is_floating_point(x))
# Tenseur d'entiers
y = torch.tensor([1, 2, 3])
print("Entier :", torch.is_floating_point(y))
# Tenseur à virgule flottante
x = torch.tensor([1.0, 2.0, 3.0])
print("Virgule flottante :", torch.is_floating_point(x))
# Tenseur d'entiers
y = torch.tensor([1, 2, 3])
print("Entier :", torch.is_floating_point(y))
Le résultat de sortie est:
浮点数: True 整数: False
Autres extensions