(prob_matrix)
return ret
Args:
        prob_matrix: probability matrix have the shape of (data_num, class_num), 
            type of float. Generated from softmax activation
            
    Returns:
        ret: return class ids, shape of(data_num,)
def pred_from_prob_single(prob_matrix):
    """

    Args:
        prob_matrix: probability matrix have the shape of (data_num, class_num), 
            type of float. Generated from softmax activation
            
    Returns:
        ret: return class ids, shape of(data_num,)
    """
    ret = np.argmax(prob_matrix, axis=1)
    return ret
def pred_from_prob_single ( prob_matrix ) : ret = np . argmax ( prob_matrix , axis = 1 ) return ret