TensorBuffer
函数功能
TensorBuffer的构造函数，用来创建TensorBuffer对象，根据传入的不同参数可选用不同的构造函数。
函数原型
TensorBuffer::TensorBuffer();
TensorBuffer::TensorBuffer(uint32_t size, MemoryData::MemoryType type, int32_t deviceId);
TensorBuffer::TensorBuffer(uint32_t size, int32_t deviceId);
TensorBuffer::TensorBuffer(uint32_t size);
TensorBuffer::TensorBuffer(void *ptr, uint32_t size, MemoryData::MemoryType type = MemoryData::MemoryType::MEMORY_HOST_NEW, int32_t deviceId = -1);
参数说明
参数名 |
输入/输出 |
说明 |
|---|---|---|
ptr |
输入 |
内存指针。 |
size |
输入 |
内存大小。 |
type |
输入 |
用于构造TensorBuffer对象的内存类型。 |
deviceId |
输入 |
用于构造TensorBuffer对象的参数，int类型数据，设备编号。 |
父主题： TensorBuffer