Batched index_select

Thanks, @eacousineau. I was looking for this.