guides/understanding_masking_and_padding.py/MyLayer/__init__
class MyLayer:    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        self.embedding = layers.Embedding(
            input_dim=5000, output_dim=16, mask_zero=True
        )
        self.lstm = layers.LSTM(32)