評價此頁

torch.mtia.set_device#

torch.mtia.set_device(device)[原始碼]#

設定當前裝置。

引數

device (torch.deviceint) – 選定的裝置。如果此引數為負數,則此函式無任何操作。