如何在Netron中看到每一层的shape
有时导出的onnx模型只有输入输出部分有shape信息,那怎么样才能让onnx记录每一层的shape呢?
只用简单的加上:
import onnx
from onnx import shape_inference
path = "..."
onnx.save(onnx.shape_inference.infer_shapes(onnx.load(path)), path)
import onnx
from onnx import shape_inference
path = "..."
onnx.save(onnx.shape_inference.infer_shapes(onnx.load(path)), path)