def ApplyPadding(input, padding, pad_value) -> torch.Tensor:
    """
    Args:
        input:   [bs, max_time_step, dim]
        padding: [bs, max_time_step]
    """
    return padding * pad_value + input * (1 - padding)
Args:
input:   [bs, max_time_step, dim]
padding: [bs, max_time_step]
modules/wenet_extractor/transducer/predictor.py