SetSize

设置tensor数据的内存大小。

void SetSize(const size_t size)

参数	输入/输出	说明
size	输入	tensor的内存大小，单位为字节。

无

无

std::vector<int> a = {10};
auto addr = reinterpret_cast<void *>(a.data());
TensorData td(addr, HostAddrManager, 100U, kOnHost);
td.SetSize(10U);