张量描述信息定义。
1 2 3 4 5
struct TensorDesc { size_t tensorSize; std::string tensorName; std::vector<int64_t> tensorDims; };
参数名
输入/输出
说明
tensorSize
输出
张量的大小,单位为Byte。
tensorName
张量名称。
tensorDims
张量维度。