比较与 torch.Tensor.cuda 的功能差异

查看源文件

torch.Tensor.cuda

torch.Tensor.cuda()

更多内容详见 torch.Tensor.cuda.

mindspore.context.set_context

mindspore.context.set_context(**kwargs)

更多内容详见 mindspore.context.set_context.

Differences

PyTorch: 将Tenosr 拷贝到 cuda 内存.

MindSpore：将变量设置为 device_target='GPU' 或 device_target='Ascend' 时, 网络和Tensor都将自动拷贝到 GPU/Ascend 设备.