Lorem ipsum dolor sit amet, consectetur adipiscing elit. In at congue neque. Sed sed iaculis turpis. Nullam pretium est nec metus mattis, ut hendrerit metus egestas. Sed faucibus nulla sed eleifend sollicitudin. Integer nec est et massa ultrices gravida at nec nulla. Sed sit amet dictum nunc. In accumsan velit viverra lectus lobortis egestas. Sed vel leo elit. Donec tempus dui id nulla sodales, nec interdum odio consequat. Mauris dictum placerat orci, in semper sapien commodo vitae. Aliquam et mauris vel tortor molestie tincidunt. Morbi vulputate, ante sit amet venenatis pretium, nibh lacus laoreet erat, id accumsan lectus est eget velit. Morbi eget finibus lectus. In hendrerit ullamcorper enim, ornare aliquet dolor sagittis vel. Etiam eu enim consectetur, convallis odio id, sollicitudin leo.

def merge_topk_batch(topk_ind, topk_dist, batch_ind, batch_dist, k):
    topk_ind[:,k:] = batch_ind[:,:k]
    topk_dist[:,k:] = batch_dist[:,:k]
    all_idx = np.argpartition(topk_dist, k, axis=1)[:,:k]
    for i, idx in enumerate(all_idx):
        topk_ind[i,:k] = topk_ind[i,idx]
        topk_dist[i,:k] = topk_dist[i,idx]
    return topk_ind, topk_dist

Quisque tincidunt porttitor mi in porttitor. Aenean vel tincidunt nisi, et malesuada nunc. In efficitur convallis porttitor. Donec enim ex, ultrices nec ullamcorper ut, finibus sit amet magna. Vivamus justo eros, euismod at purus vel, tristique mollis justo. Donec et augue condimentum, aliquam felis quis, tristique velit. Praesent at enim eu purus gravida mattis vel tristique felis. Pellentesque rutrum non libero in congue. Vestibulum felis lorem, elementum sed sapien vel, sagittis accumsan odio. Aliquam iaculis feugiat felis. Curabitur id risus quam. Etiam finibus ut erat sit amet posuere. In lobortis ipsum eget est laoreet porttitor.