torch.get_device_module

torch.get_device_module(device=None)

返回與給定裝置關聯的模組（例如，torch.device(‘cuda’)，“mtia:0”，“xpu”，...）。如果沒有給出裝置，則返回當前加速器或 CPU（如果不存在）的模組。