added batch_axis parameter for CrossEntropy losses

src/main/grammars/de/monticore/lang/monticar/CNNTrain.mc4

@@ -108,6 +108,7 @@ grammar CNNTrain extends de.monticore.lang.monticar.Common2, de.monticore.Number
     SparseLabelEntry implements CrossEntropyEntry, SoftmaxCrossEntropyEntry, SoftmaxCrossEntropyIgnoreIndicesEntry = name:"sparse_label" ":" value:BooleanValue;
     FromLogitsEntry implements SoftmaxCrossEntropyEntry, SoftmaxCrossEntropyIgnoreIndicesEntry, KullbackLeiblerEntry = name:"from_logits" ":" value:BooleanValue;
     LossAxisEntry implements CrossEntropyEntry, SoftmaxCrossEntropyEntry, SoftmaxCrossEntropyIgnoreIndicesEntry = name:"loss_axis" ":" value:IntegerValue;
+    BatchAxisEntry implements CrossEntropyEntry, SoftmaxCrossEntropyEntry, SoftmaxCrossEntropyIgnoreIndicesEntry = name:"batch_axis" ":" value:IntegerValue;
     IgnoreIndicesEntry implements SoftmaxCrossEntropyIgnoreIndicesEntry = name:"ignore_indices" ":" value:IntegerValue;
     MarginEntry implements HingeEntry, SquaredHingeEntry = name:"margin" ":" value:NumberValue;
     LabelFormatEntry implements LogisticEntry = name:"label_format" ":" value:StringValue;

src/main/java/de/monticore/lang/monticar/cnntrain/_cocos/ParameterAlgorithmMapping.java

@@ -48,6 +48,7 @@ class ParameterAlgorithmMapping {
 		ASTLossWeightsEntry.class,
         ASTSparseLabelEntry.class,
         ASTLossAxisEntry.class,
+        ASTBatchAxisEntry.class,
         ASTFromLogitsEntry.class,
         ASTIgnoreIndicesEntry.class,
         ASTMarginEntry.class,

src/test/resources/valid_tests/FullConfig.cnnt

@@ -8,6 +8,7 @@ configuration FullConfig{
         sparse_label: true
         from_logits: true
         loss_axis : -1
+        batch_axis: 0
      }
      context : gpu
      normalize : true