(s_sc, beam_size)
return pr_sc_beam
return: [ pr_wc1_i, pr_wc2_i, ...]
return: [ pr_wc1_i, pr_wc2_i, ...]
return : [ pr_wc1_i pr_wc2_i ... ]
def pred_sc_beam(s_sc, beam_size):
    """
    return: [ pr_wc1_i, pr_wc2_i, ...]
    """
    # get g_num
    pr_sc_beam = []


    for s_sc1 in s_sc:
        val, idxes = s_sc1.topk(k=beam_size)
        pr_sc_beam.append(idxes.tolist())

    return pr_sc_beam
def pred_sc_beam ( s_sc , beam_size ) : # get g_num pr_sc_beam = [ ] for s_sc1 in s_sc : val , idxes = s_sc1 . topk ( k = beam_size ) pr_sc_beam . append ( idxes . tolist ( ) ) return pr_sc_beam