to_device
函数功能
搬移image到Device内存中。
函数原型
1 | to_device(deviceId: int) |
输入参数说明
参数名 |
类型 |
说明 |
|---|---|---|
deviceId |
int |
Device设备的ID号。 |
抛异常接口
调用Tensor的to_device函数失败,抛出runtime异常。
父主题: Image
搬移image到Device内存中。
1 | to_device(deviceId: int) |
参数名 |
类型 |
说明 |
|---|---|---|
deviceId |
int |
Device设备的ID号。 |
调用Tensor的to_device函数失败,抛出runtime异常。