(s_wc)
return pr_wc
return: [ pr_wc1_i, pr_wc2_i, ...]
    ! Returned index is sorted by prob.
    All colume-indexes are returned here.
return: [ pr_wc1_i, pr_wc2_i, ...]
    ! Returned index is sorted by prob.
    All colume-indexes are returned here.
return : [ pr_wc1_i pr_wc2_i ... ] ! Returned index is sorted by prob . All colume - indexes are returned here .
def pred_wc_sorted_by_prob(s_wc):
    """
    return: [ pr_wc1_i, pr_wc2_i, ...]
    ! Returned index is sorted by prob.
    All colume-indexes are returned here.
    """
    # get g_num
    bS = len(s_wc)
    pr_wc = []

    for b in range(bS):
        s_wc1 = s_wc[b]
        pr_wc1 = argsort(-s_wc1.data.cpu().numpy())
        pr_wc.append(list(pr_wc1))
    return pr_wc
def pred_wc_sorted_by_prob ( s_wc ) : # get g_num bS = len ( s_wc ) pr_wc = [ ] for b in range ( bS ) : s_wc1 = s_wc [ b ] pr_wc1 = argsort ( - s_wc1 . data . cpu ( ) . numpy ( ) ) pr_wc . append ( list ( pr_wc1 ) ) return pr_wc