import torch


def get_padding_mask(input_ids: torch.Tensor, padding_token_id: int) -> torch.Tensor:
    """

    :param input_ids: tensor of shape [L, B]
    :param padding_token_id: id of the padding token
    """
    return (input_ids.T == padding_token_id).bool()