Function mse_loss_with_reduction 

pub fn mse_loss_with_reduction<B>(
    prediction: &Tensor<B>,
    target: &Tensor<B>,
    reduction: Reduction,
) -> Result<Tensor<B>, Error>
where
    B: Backend,

MSE Loss with configurable reduction.