4b81ac641f68efc4c7e26ef03093781e28894c14,gluon/gluoncv2/models/superpointnet.py,SPDescriptor,hybrid_forward,#SPDescriptor#Any#Any#Any#,202

Before Change

        pts_tr = pts.transpose(axes=(0, 2, 1))
        for i in range(batch_size):
            desc_map_i = desc_map[i]
            pts_tr_i = pts_tr[i].reshape(shape=(2, -1))
            pts_ravel_i = F.ravel_multi_index(pts_tr_i, shape=in_size)
            desc_map_sorted_i = F.take(desc_map_i, pts_ravel_i)
            desc_map_sorted_list.append(desc_map_sorted_i)
        desc_map_sorted = F.stack(*desc_map_sorted_list)

After Change

        desc_list = []
        for i in range(batch_size):
            desc_map_i = desc_map[i]
            pts_i_tr = pts[i].transpose()
            pts_ravel_i = F.ravel_multi_index(pts_i_tr, shape=in_size)
            desc_map_sorted_i = F.take(desc_map_i, pts_ravel_i)
            desc_list.append(desc_map_sorted_i)

Instances