PyTorch torch.get_default_device Function

Pytorch torch 参考手册PyTorch torch Reference Manual

torch.get_default_deviceIs a function in PyTorch used to get the current default device.

Function Definition

torch.get_default_device()

Usage Example

Example

import torch

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

Pytorch torch 参考手册PyTorch torch Reference Manual

Other Extensions