Add parsing for logsoftmax

mldiego · Mar 15, 2024 · cfa086a · cfa086a
1 parent 30b6aee
commit cfa086a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/code/nnv/engine/utils/matlab2nnv.m b/code/nnv/engine/utils/matlab2nnv.m
@@ -44,7 +44,7 @@
     % Layers with no effect on the reachability analysis
     if isa(L, 'nnet.cnn.layer.DropoutLayer') || isa(L, 'nnet.cnn.layer.SoftmaxLayer') || isa(L, 'nnet.cnn.layer.ClassificationOutputLayer') ...
             || isa(L,"nnet.onnx.layer.VerifyBatchSizeLayer") || isa(L, "nnet.cnn.layer.RegressionOutputLayer") ...
-            || customLayer_no_NLP == 1 || isa(L, "nnet.onnx.layer.CustomOutputLayer")
+            || customLayer_no_NLP == 1 || isa(L, "nnet.onnx.layer.CustomOutputLayer") || contains(class(L), "LogSoftmax")
         Li = PlaceholderLayer.parse(L);
 
     elseif customLayer_no_NLP