torch.get_device_module

torch.get_device_module(device=None)[源代码]

根据给定的设备（如 torch.device('cuda')、"mtia:0"、"xpu" 等）返回相应的模块。如果没有指定设备，则返回当前加速器的模块；若无加速器存在，则返回 CPU 的模块。