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