def _create_selfattn_mask(self, x):
    bsz = x.size(0)
    time = x.size(1)
    mask = torch.tril(x.new(time, time).fill_(1))
    mask = mask.unsqueeze(0).expand(bsz, (- 1), (- 1))
    return mask
def _create_selfattn_mask(self, x):
    bsz = x.size(0)
    time = x.size(1)
    mask = torch.tril(x.new(time, time).fill_(1))
    mask = mask.unsqueeze(0).expand(bsz, (- 1), (- 1))
    return mask