torch.mtia.set_device¶
- torch.mtia.set_device(device)[source]¶
设置当前设备。
- 参数
device (torch.device or int) – 选定的设备。如果此参数为负，则此函数为无操作。
- 文档 >
- torch.mtia >
- torch.mtia.set_device
-
快捷方式
设置当前设备。
device (torch.device or int) – 选定的设备。如果此参数为负，则此函数为无操作。
为了分析流量并优化您的体验，我们在本网站上提供 cookie。点击或浏览表示您同意我们使用 cookie。作为本网站的当前维护者，Facebook 的 Cookie 政策适用。了解详情，包括关于可用控制的信息：Cookie 政策.
