ToDevice

函数功能

将Tensor数据转移到Device侧。
  • 如果原始的内存为用户通过构造函数传入,则该原始内存由用户自行管理释放。
  • 若是通过TensorMalloc或者Malloc申请的内存,则是将数据转移到Device侧,原始内存被自动释放。

函数原型

1
APP_ERROR Tensor::ToDevice(int32_t deviceId);

参数说明

参数名

输入/输出

说明

deviceId

输入

将Tensor转移到deviceId对应的设备。

“deviceId”需为有效的设备ID。

返回参数说明

数据结构

说明

APP_ERROR

程序执行返回的错误码,请参考APP_ERROR说明