Tensor

函数功能

Tensor类的构造函数。

函数原型

Tensor()   //默认构造函数。 构造失败时,会抛出std::runtime_error异常
Tensor(const Tensor &tensor); // 支持拷贝构造
Tensor(const std::vector<uint32_t> &shape, const MxBase::TensorDType &dataType, const int32_t &deviceId = -1)
//不传入内存,可搭配TensorMalloc接口进行内存申请,申请的内存无需用户管理释放。 构造失败时,会抛出std::runtime_error异常
Tensor(void* usrData,const std::vector<uint32_t> &shape, const MxBase::TensorDType &dataType, const int32_t &deviceId = -1)
//传入用户自己构造的数据内存,需要用户自身对此内存进行管理(保证内存数据的生命周期)。 构造失败时,会抛出std::runtime_error异常

参数说明

参数名

输入/输出

说明

tensor

输入

已初始化的其它Tensor。

usrData

输入

用户构造的输入内存,该内存由用户管理申请和释放。

shape

输入

Tensor的shape属性。

dataType

输入

Tensor的数据类型,具体请参见TensorDType

deviceId

输入

Tensor所在的设备ID,默认为-1,在Host侧。