to_device

函数功能

搬移image到Device内存中。

函数原型

1
to_device(deviceId: int)

输入参数说明

参数名

类型

说明

deviceId

int

Device设备的ID号。

抛异常接口

调用Tensor的to_device函数失败,抛出runtime异常。