[GitHub] apeforest commented on a change in pull request #12176: MXPredReshape bug: need to reshape softmax_label

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

apeforest commented on a change in pull request #12176: MXPredReshape bug: need to reshape softmax_label
URL: https://github.com/apache/incubator-mxnet/pull/12176#discussion_r221319975
 
 

 ##########
 File path: src/c_api/c_predict_api.cc
 ##########
 @@ -304,7 +304,8 @@ int MXPredReshape(mx_uint num_input_nodes,
   for (size_t i=0; i < arg_names.size(); ++i) {
     TShape newShape = arg_shapes[i];
     NDArray &arr = p->arg_arrays[i];
-    if (new_shape.count(arg_names[i]) != 0) {
+    if (new_shape.count(arg_names[i]) != 0 ||
+        strcmp("softmax_label", arg_names[i].c_str()) == 0) {
 
 Review comment:
   Still trying to understand the problem. But this hack is not acceptable. 

----------------------------------------------------------------
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