Fonction torch.get_default_device de PyTorch

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

torch.get_default_deviceEst une fonction dans PyTorch utilisée pour obtenir le périphérique par défaut actuel.

Définition de fonction

torch.get_default_device()

Exemple d'utilisation

Exemple

import torch

device = torch.get_default_device()
print("Default device:", device)

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

Autres extensions