Fonction torch.can_cast de PyTorch

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

torch.can_castIl s'agit d'une fonction de PyTorch utilisée pour vérifier si une conversion de type peut être effectuée. Elle vérifie si la conversion du type de données source vers le type de données cible est possible.

Définition de fonction

torch.can_cast(from_dtype, to_dtype)

Exemple d'utilisation

Exemple

import torch

# Vérifier si la conversion de type peut être effectuée
print(torch.can_cast(torch.float32, torch.int32))   # Vrai
print(torch.can_cast(torch.float64, torch.int8))    # True
print(torch.can_cast(torch.int64, torch.bool))      # True

# Convertir depuis bool
print(torch.can_cast(torch.bool, torch.float32))    # True

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

Autres extensions