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