ToDevice
函数功能
将Tensor数据转移到Device侧。
- 如果原始的内存为用户通过构造函数传入，则该原始内存由用户自行管理释放。
- 若是通过TensorMalloc或者Malloc申请的内存，则是将数据转移到Device侧，原始内存被自动释放。
函数原型
APP_ERROR Tensor::ToDevice(int32_t deviceId);
参数说明
参数名 |
输入/输出 |
说明 |
|---|---|---|
deviceId |
输入 |
将Tensor转移到deviceId对应的设备。 “deviceId”需为有效的设备ID。 |
返回参数说明
数据结构 |
说明 |
|---|---|
APP_ERROR |
程序执行返回的错误码，请参考APP_ERROR说明。 |
父主题： Tensor