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