BaseTensor
功能
定义一个张量数据的结构。
结构定义
1 2 3 4 5 | struct BaseTensor { void* buf; std::vector<int> shape; size_t size; } |
参数说明
参数名 |
说明 |
---|---|
buf |
张量的数据。 |
shape |
张量的形状。 |
size |
张量数据对应内存大小,单位为Byte。 size大小应与实际内存大小一致,否则可能会导致程序出现coredump情况。 |
父主题: 通用数据结构