def get_masks(tokens, max_seq_len):
    """Mask for padding"""
    if len(tokens) > max_seq_len:
        raise IndexError('Token length more than max seq length!')
    return [1] * len(tokens) + [0] * (max_seq_len - len(tokens))
