[GitHub] GeneZC opened a new issue #12433: How to use multiple output nodes in cpp

[GitBox <gi...@apache.org>]

GeneZC opened a new issue #12433: How to use multiple output nodes in cpp
URL: https://github.com/apache/incubator-mxnet/issues/12433
 
 
   Using gluon, I've created a net similar to ResNet used in AlphaGo Zero (there are two output nodes processed by policy head and value head) I just return two results in one line like below:
   `return policy, value`
   Then I export the model and there are '.json' for the architecture and '.params' for parameters.
   After that, I use API in CPP to do inference, however, I can get two different results trough setting different index in function
   ` int MXPredGetOutputShape(PredictorHandle handle,
                                      mx_uint index,
                                      mx_uint** shape_data,
                                      mx_uint* shape_ndim); ` 
   

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services