Функция torch.get_default_device PyTorch

Pytorch torch 参考手册Справочное руководство PyTorch torch

torch.get_default_deviceЭто функция в PyTorch, используемая для получения текущего устройства по умолчанию.

Определение функции

torch.get_default_device()

Пример использования

Пример

import torch

device = torch.get_default_device()
print("Устройство по умолчанию:", device)

Pytorch torch 参考手册Справочное руководство PyTorch torch

Другие расширения