比较与 torch.Tensor.cuda 的功能差异
torch.Tensor.cuda
torch.Tensor.cuda()
更多内容详见 torch.Tensor.cuda.
mindspore.set_context
mindspore.set_context(**kwargs)
更多内容详见 mindspore.set_context.
Differences
PyTorch: 将Tenosr 拷贝到 cuda 内存.
MindSpore:将变量设置为 device_target='GPU'
或 device_target='Ascend'
时, 网络和Tensor都将自动拷贝到 GPU/Ascend 设备.