更新时间:2021-03-18 GMT+08:00
分享

AITensorDescription

tensor描述,主要用于描述模型的输入输出信息,详细实现参考ai_types.proto。

// Tensor描述
message AITensorDescription
{
    string name = 1;   // Tensor名称
    string type = 2;   // Tensor类型
    repeated string compatible_type = 3; // 指定可以兼容的所有父类类型
    repeated AITensorParaDescription paras = 4; // 参数列表
};
分享:

    相关文档

    相关产品