昇腾社区首页
中文
注册

BaseTensor

功能

定义一个张量数据的结构。

结构定义

struct BaseTensor { 
     void* buf; 
     std::vector<int> shape; 
     size_t size; 
 }

参数说明

参数名

输入/输出

说明

buf

输入

张量的数据。

shape

输入

张量的形状。

size

输入

张量数据对应内存大小,单位为Byte。

size大小应与实际内存大小一致,否则可能会导致程序出现coredump情况。