pub fn apply_mask(
log_probs: &[Vec<Vec<f64>>],
masks: &[Vec<Vec<bool>>],
) -> Vec<Vec<Vec<f64>>>Expand description
Applies a boolean constraint mask to log-probabilities.
Invalid tokens (mask == false) are set to f64::NEG_INFINITY.
Input / output shape: [B][M][|V|].