设置tensor数据的内存大小。
void SetSize(const size_t size)
参数
输入/输出
说明
size
输入
tensor的内存大小,单位为字节。
无。
auto addr = reinterpret_cast<void *>(0x10); TensorData td(addr, HostAddrManager, 100U, kOnHost); td.SetSize(10U);