(wn, s_wc)
return pr_wc
return: [ pr_wc1_i, pr_wc2_i, ...]
    ! Returned index is sorted!
return: [ pr_wc1_i, pr_wc2_i, ...]
    ! Returned index is sorted!
return : [ pr_wc1_i pr_wc2_i ... ] ! Returned index is sorted!
def pred_wc(wn, s_wc):
    """
    return: [ pr_wc1_i, pr_wc2_i, ...]
    ! Returned index is sorted!
    """
    # get g_num
    pr_wc = []
    for b, wn1 in enumerate(wn):
        s_wc1 = s_wc[b]

        pr_wc1 = argsort(-s_wc1.data.cpu().numpy())[:wn1]
        pr_wc1.sort()

        pr_wc.append(list(pr_wc1))
    return pr_wc
def pred_wc ( wn , s_wc ) : # get g_num pr_wc = [ ] for b , wn1 in enumerate ( wn ) : s_wc1 = s_wc [ b ] pr_wc1 = argsort ( - s_wc1 . data . cpu ( ) . numpy ( ) ) [ : wn1 ] pr_wc1 . sort ( ) pr_wc . append ( list ( pr_wc1 ) ) return pr_wc