Added mask option to DotProductSelfAttention layer

parent e43ddc36
Pipeline #310121 failed with stage
in 38 seconds
......@@ -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();
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment