Added mask option to DotProductSelfAttention layer

e73431ac · Julian Johannes Steinsberger-Dührßen · e43ddc36 · e73431ac
Commit e73431ac authored Jul 20, 2020 by Julian Johannes Steinsberger-Dührßen
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

src/main/java/de/monticore/lang/monticar/cnnarch/generator/ArchitectureElementData.java ...g/monticar/cnnarch/generator/ArchitectureElementData.java +4 -0

No files found.
--- a/src/main/java/de/monticore/lang/monticar/cnnarch/generator/ArchitectureElementData.java
+++ b/src/main/java/de/monticore/lang/monticar/cnnarch/generator/ArchitectureElementData.java
@@ -215,6 +215,10 @@ public class ArchitectureElementData {
        return getLayerSymbol().getBooleanValue(AllPredefinedLayers.USE_PROJ_BIAS_NAME).get();
    }

+    public boolean getUseMask() {
+        return getLayerSymbol().getBooleanValue(AllPredefinedLayers.USE_MASK_NAME).get();
+    }
+
    public int getSubKeySize(){
        return getLayerSymbol().getIntValue(AllPredefinedLayers.SUB_KEY_SIZE_NAME).get();
    }