Vectorized version of get_local_energy. Takes similar arguments as get_local_energy but with additional array axes over which get_local_energy is mapped.