如何获取onnx神经网络输出层的维度?
我可以得到 onnx 图,但没有输出尺寸:
~/onnx-tensorrt/third_party/onnx/onnx/tools/net_drawer.py --input ./weights/tiny_3l_v5_11_608.onnx --output ./weights/tiny_3l_v5_11_608.dot --embed_docstring
Run Code Online (Sandbox Code Playgroud)
特尔维辛,马库斯