torch.get_device_module

torch.get_device_module(device=None)

傳回與給定裝置相關聯的模組 (例如，torch.device('cuda'), “mtia:0”, “xpu”, …)。如果沒有給定裝置，則傳回目前加速器的模組；如果沒有加速器，則傳回 CPU 的模組。