获取tensor数据的内存大小
size_t GetSize() const
无。
tensor所占内存大小,单位为字节。
auto addr = reinterpret_cast<void *>(0x10); TensorData td(addr, HostAddrManager, 100U, kOnHost); auto td_size = td.GetSize(); // 100U