TensorBase的构造函数,用来创建TensorBase对象,根据传入的不同参数可选用不同的构造函数。
如出现因内存耗尽,导致构造函数执行失败抛出的异常时,请勿继续调用后续的成员函数。
原型1:
1 | TensorBase::TensorBase(); |
原型2:
1 | TensorBase::TensorBase(const MemoryData &memoryData, const bool &isBorrowed, const std::vector<uint32_t> &shape, const TensorDataType &type); |
原型3:
TensorBase::TensorBase(const std::vector<uint32_t> &shape, const TensorDataType &type, const MemoryData::MemoryType &bufferType, const int32_t &deviceId);
原型4:
1 | TensorBase::TensorBase(const std::vector<uint32_t> &shape, const TensorDataType &type, const int32_t &deviceId); |
原型5:
1 | TensorBase::TensorBase(const std::vector<uint32_t> &shape, const TensorDataType &type); |
原型6:
1 | TensorBase::TensorBase(const std::vector<uint32_t> &shape); |
原型7:
1 | TensorBase::TensorBase(const TensorBase& tensor) = default; |
参数名 |
输入/输出 |
说明 |
---|---|---|
memoryData |
输入 |
用于构造TensorBase对象的参数,内存管理结构体,具体请参见MemoryData。 |
isBorrowed |
输入 |
表示传入的MemoryData数据是否要Tensor主动释放。
|
shape |
输入 |
用于构造TensorBase对象的参数,张量的形状。 |
type |
输入 |
用于构造TensorBase对象的参数,TensorDataType类型数据,具体请参见TensorDataType枚举说明。 |
bufferType |
输入 |
用于构造TensorBase对象的参数,张量数据的内存类型。 |
deviceId |
输入 |
用于构造TensorBase对象的参数,int类型数据,设备编号。 |
tensor |
输入 |
TensorBase对象,用于构造TensorBase对象的参数。 |