Im getting as mask:
tensor([[False],
        [False],
        [False],
        [False],
        [True],
        [True]], device='cuda:0')

of shape (batch_size, target_length), where False is a non-padding token and True marks padding tokens. Is that correct?