Fonction torch.is_storage de PyTorch
Manuel de référence Pytorch torch
torch.is_storageIl s'agit d'une fonction dans PyTorch utilisée pour vérifier si un objet est un objet de stockage.
Définition de fonction
torch.is_storage(obj)
Exemple d'utilisation
Exemple
import torch
# Créer un tenseur
x = torch.tensor([1, 2, 3])
# Vérifier s'il s'agit d'un objet de stockage
print(torch.is_storage(x))
print(torch.is_storage(x.storage()))
# Créer un tenseur
x = torch.tensor([1, 2, 3])
# Vérifier s'il s'agit d'un objet de stockage
print(torch.is_storage(x))
print(torch.is_storage(x.storage()))
Le résultat de sortie est:
False True
Autres extensions