Fonction torch.set_default_dtype de PyTorch

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

torch.set_default_dtypeIl s'agit d'une fonction de PyTorch utilisée pour définir le type de données à virgule flottante par défaut.

Définition de fonction

torch.set_default_dtype(d)

Exemple d'utilisation

Exemple

import torch

# Définir le dtype par défaut sur float64
torch.set_default_dtype(torch.float64)

# Créer un tenseur, en utilisant float64 par défaut
x = torch.tensor([1.0, 2.0, 3.0])
print("Default dtype:", x.dtype)

# Rétablir la valeur par défaut
torch.set_default_dtype(torch.float32)

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

Autres extensions