guides/understanding_masking_and_padding.py/MyActivation/__init__
class MyActivation:    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        # Signal that the layer is safe for mask propagation
        self.supports_masking = True