fixed problem with expected size of argmax outputs being too high in C++ code

src/main/java/de/monticore/lang/monticar/cnnarch/gluongenerator/CNNArch2GluonTemplateController.java

@@ -386,6 +386,10 @@ public class CNNArch2GluonTemplateController extends CNNArchTemplateController {
         return AllAttentionModels.getAttentionModels().contains(getComponentName());
     }
 
+    public boolean isArchitectureOutput(String element){
+        return getArchitectureOutputs().contains(element.replace("1000000", "0"));
+    }
+
     public int getBeamSearchMaxLength(UnrollInstructionSymbol unroll){
         return unroll.getIntValue(AllPredefinedLayers.MAX_LENGTH_NAME).get();
     }

src/main/resources/templates/gluon/CNNPredictor.ftl

@@ -52,7 +52,10 @@ public:
         output_index = ${variable?index?c};
         MXPredGetOutputShape(handle, output_index, &shape, &shape_len);
         size = 1;
+<#if !(tc.isArchitectureOutput(variable) && tc.endsWithArgmax(networkInstruction.body))>
         for (mx_uint i = 0; i < shape_len; ++i) size *= shape[i];
+</#if>
+
         assert(size == out_${variable}.size());
         MXPredGetOutput(handle, output_index, &(out_${variable}[0]), out_${variable}.size());