Method test_relu

tests/python/test_kernel.py:103–108

(self)

class TestWhere:
    def test_relu(self) -> None:
        x = kernel.arg("x")
        relu = kernel.elementwise(kernel.where(x > 0, x, 0))
        data = rt.asarray(np.array([-2.0, -1.0, 0.0, 1.0, 2.0]))
        out = rt.go(kernel.map(relu, x=data)).result().numpy()
        np.testing.assert_array_equal(out, [0.0, 0.0, 0.0, 1.0, 2.0])


class TestReduce: