Pytorch中的torch.device该如何迁移到MindSpore中

想把一个用PyTorch写的UNet网络转成用MindSpore写的，其中一个torch.device的API找不到MindSpore的映射，请问应该怎么改？

torch写的代码的运行结果如下：

​

尝试用ms.set_context(device_target="GPU\cpu\acsend")，但是运行结果输出为None.

****************************************************解答*****************************************************

MindSpore的求导类似pytorch的 torch.autograd.grad（链接torch.autograd.grad — PyTorch 1.12 documentation），将梯度与Tensor剥离，grad操作直接返回梯度，因此不需要 optimizer.zero_grad(), loss.backward() 这两个步骤。 

由于 Tensor与梯度剥离，无法直接通过 optimizer.step() 进行参数更新，因此，在MindSpore中更新参数的方式为 optimizer(grads)，可以参考 https://www.mindspore.cn/docs/api/zh-CN/r1.5/_modules/mindspore/nn/wrap/cell_wrapper.html#TrainOneStepCell 中construct里的写法。