Fonction torch.get_default_dtype de PyTorch

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

torch.get_default_dtypeCeci est une fonction dans PyTorch utilisée pour obtenir le type de données à virgule flottante par défaut actuel.

Définition de fonction

torch.get_default_dtype()

Exemple d'utilisation

Exemple

import torch

# Obtenir le dtype par défaut
dtype = torch.get_default_dtype()
print("Default dtype:", dtype)

Le résultat de sortie est:

默认 dtype: torch.float32

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

Autres extensions