guides/understanding_masking_and_padding.py/CustomEmbedding/compute_mask
class CustomEmbedding:    def compute_mask(self, inputs, mask=None):
        if not self.mask_zero:
            return None
        return ops.not_equal(inputs, 0)