def _(then): if then: qy.return_(-numpy.inf) else: qy.return_(a + qy.log1p(qy.exp(qy.select(s, y_in, x_in) - a)))
def _(k): posterior_pi = out.at(k).data.gep(0, 0) normalized_pi = posterior_pi.load() - total_value qy.exp(normalized_pi).store(posterior_pi)
def _(k): responsibility = r_KN.at(k, n).data qy.exp(responsibility.load() - total_value).store(responsibility)
def _(): v0 = qy.exp(1.0) @qy.python(v0) def _(v0_py): assert_equal(v0_py, math.e)