pub fn entanglement_entropy_batch( states: &[Vec<Complex64>], subsystem_dimension: usize, environment_dimension: usize, order: EntropyOrder, ) -> Result<Vec<f64>>