def test_builder_eval(self): x_val = np.array([[-1, 2, -3], [4, -5, 6]], dtype=np.float32) v = mb.clamped_relu(x=x_val, alpha=2.0, beta=1.0) x = np.minimum(np.maximum(x_val, 0), 1.0) y = np.minimum(np.minimum(x_val, 0) * 2.0, 1.0) np.testing.assert_allclose(x + y, v.val, atol=1e-04, rtol=1e-05)
def build(x): return [mb.clamped_relu(x=x, alpha=alpha, beta=beta)]
def build(x): return mb.clamped_relu(x=x, alpha=2.0, beta=1.0)