generate_index_benchmarks: add support for shape specification