(s_sc)
return pr_sc
return: [ pr_wc1_i, pr_wc2_i, ...]
return: [ pr_wc1_i, pr_wc2_i, ...]
return : [ pr_wc1_i pr_wc2_i ... ]
def pred_sc(s_sc):
    """
    return: [ pr_wc1_i, pr_wc2_i, ...]
    """
    # get g_num
    pr_sc = []
    for s_sc1 in s_sc:
        pr_sc.append(s_sc1.argmax().item())

    return pr_sc
def pred_sc ( s_sc ) : # get g_num pr_sc = [ ] for s_sc1 in s_sc : pr_sc . append ( s_sc1 . argmax ( ) . item ( ) ) return pr_sc