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