def compute_logits(theta, ob):
    """
    theta: A matrix of size |A| * (|S|+1)
    ob: A vector of size |S|
    return: A vector of size |A|
    """
    #ob_1 = include_bias(ob)
    logits = ob.dot(theta.T)
    return logits