Updated tests

src/test/resources/target_code/gluon/CNNPredictor_mnist_mnistClassifier_net.h

@@ -44,7 +44,7 @@ public:
         MXPredGetOutputShape(handle, output_index, &shape, &shape_len);
         size = 1;
         for (mx_uint i = 0; i < shape_len; ++i) size *= shape[i];
-        assert(size == out_predictions_.size());
+        assert(out_predictions_.size() == 1 || size == out_predictions_.size());
         MXPredGetOutput(handle, output_index, &(out_predictions_[0]), out_predictions_.size());
 
     }

src/test/resources/target_code/gluon/reinforcementModel/cartpole/CNNPredictor_cartpole_master_dqn.h

@@ -44,7 +44,7 @@ public:
         MXPredGetOutputShape(handle, output_index, &shape, &shape_len);
         size = 1;
         for (mx_uint i = 0; i < shape_len; ++i) size *= shape[i];
-        assert(size == out_qvalues_.size());
+        assert(out_qvalues_.size() == 1 || size == out_qvalues_.size());
         MXPredGetOutput(handle, output_index, &(out_qvalues_[0]), out_qvalues_.size());
 
     }

src/test/resources/target_code/gluon/reinforcementModel/mountaincar/CNNPredictor_mountaincar_master_actor.h

@@ -44,7 +44,7 @@ public:
         MXPredGetOutputShape(handle, output_index, &shape, &shape_len);
         size = 1;
         for (mx_uint i = 0; i < shape_len; ++i) size *= shape[i];
-        assert(size == out_action_.size());
+        assert(out_action_.size() == 1 || size == out_action_.size());
         MXPredGetOutput(handle, output_index, &(out_action_[0]), out_action_.size());
 
     }

src/test/resources/target_code/gluon/reinforcementModel/torcs/CNNPredictor_torcs_agent_torcsAgent_dqn.h

@@ -44,7 +44,7 @@ public:
         MXPredGetOutputShape(handle, output_index, &shape, &shape_len);
         size = 1;
         for (mx_uint i = 0; i < shape_len; ++i) size *= shape[i];
-        assert(size == out_qvalues_.size());
+        assert(out_qvalues_.size() == 1 || size == out_qvalues_.size());
         MXPredGetOutput(handle, output_index, &(out_qvalues_[0]), out_qvalues_.size());
 
     }

src/test/resources/target_code/gluon/reinforcementModel/torcs_td3/CNNPredictor_torcs_agent_torcsAgent_actor.h

@@ -44,7 +44,7 @@ public:
         MXPredGetOutputShape(handle, output_index, &shape, &shape_len);
         size = 1;
         for (mx_uint i = 0; i < shape_len; ++i) size *= shape[i];
-        assert(size == out_commands_.size());
+        assert(out_commands_.size() == 1 || size == out_commands_.size());
         MXPredGetOutput(handle, output_index, &(out_commands_[0]), out_commands_.size());
 
     }