Function apply_pruning_mask 

pub fn apply_pruning_mask(
    weights: &Tensor,
    mask: &Tensor,
) -> Result<Tensor, ModelError>

Apply a binary mask to weights (element-wise multiply).