def test_switch(self): val = np.random.random() xth = KTH.variable(val) xth = KTH.switch(xth >= 0.5, xth * 0.1, xth * 0.2) xtf = KTF.variable(val) xtf = KTF.switch(xtf >= 0.5, xtf * 0.1, xtf * 0.2) zth = KTH.eval(xth) ztf = KTF.eval(xtf) assert zth.shape == ztf.shape assert_allclose(zth, ztf, atol=1e-05) xth1 = KTH.variable(0.7) xth2 = KTH.variable([1, 0.2]) with pytest.raises(ValueError): xth = KTH.switch(xth1 >= 0.5, xth2 * 0.1, xth2 * 0.2) xth = KTH.switch(xth2 >= 0.5, xth1 * 0.1, xth1 * 0.2) assert_allclose(xth, KTH.variable([0.1, 0.2 * 0.2]), atol=1e-05)
def test_switch(self): val = np.random.random() xth = KTH.variable(val) xth = KTH.switch(xth >= 0.5, xth * 0.1, xth * 0.2) xtf = KTF.variable(val) xtf = KTF.switch(xtf >= 0.5, xtf * 0.1, xtf * 0.2) zth = KTH.eval(xth) ztf = KTF.eval(xtf) assert zth.shape == ztf.shape assert_allclose(zth, ztf, atol=1e-05)