Hi,
How to get output shape directly from the loaded model after run? Is it possible to get output directly without copying to empty numpy array?
Ex: The output shapes of my model tested in keras are (1,300, 4), (1,300,3,25), (1,300)
how to get output shape?
Thanks!!
1 Like
As of now there is no direct API to query number of outputs and the shapes. But, there is an issue in discussion to support it https://github.com/dmlc/tvm/issues/1454