比较与 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 设备.