标签: python python-3.x model neural-network tensor
要完成我的大学项目,我需要通过python从ONNX模型中收集以下统计信息。如果您能帮助我,我将非常感激。 找到每个层/操作员的编号, 计算CONV运算符的内核大小,步幅,输入/输出张量大小, 计算重量数据的总大小,该数据对于推理工作负载是只读的, 查找运行时产生和消耗的数据。
对于执行模型,我使用ONNX-Runtime
提前谢谢。