PyTorch torch.get_default_device 関数

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 リファレンスマニュアル

その他の拡張