下载
中文
注册

TensorDesc

功能

张量描述信息定义。

结构定义

struct TensorDesc { 
    size_t tensorSize; 
    std::string tensorName; 
    std::vector<int64_t> tensorDims; 
};

参数说明

参数名

输入/输出

说明

tensorSize

输出

张量的大小,单位为Byte。

tensorName

输出

张量名称。

tensorDims

输出

张量维度。