PyTorch torch.get_default_device función

Pytorch torch 参考手册Pytorch torch manual de referencia

torch.get_default_deviceEs una función en PyTorch que se utiliza para obtener el dispositivo predeterminado actual.

Definición de la función

torch.get_default_device()

Ejemplo de uso

Ejemplo

import torch

device = torch.get_default_device()
print("Dispositivo predeterminado:", device)

Pytorch torch 参考手册Pytorch torch manual de referencia

Otras extensiones