to_device

函数功能

移动Tensor到Device内存中。

函数原型

to_device(deviceId: int)

输入参数说明

抛异常接口

分配Tensor内存失败及拷贝到Device侧失败，抛出runtime异常。