pub fn posterior_mean_median( theta: &Array2<f64>, psi: &Array2<f64>, w: &Array1<f64>, ) -> Result<(Array2<f64>, Array2<f64>)>