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