guides/understanding_masking_and_padding.py/CustomEmbedding/__init__
class CustomEmbedding:    def __init__(self, input_dim, output_dim, mask_zero=False, **kwargs):
        super().__init__(**kwargs)
        self.input_dim = input_dim
        self.output_dim = output_dim
        self.mask_zero = mask_zero