compute_batched_distance_matrix function

Computes the distance matrix in a batched manner to save memory.