def test_hard_thresh_cuda(self): x = cp.array([-2, -1.5, -1, 0.5, 0, 0.5, 1, 1.5, 2]) y = cp.array([-2, -1.5, 0, 0, 0, 0, 0, 1.5, 2]) lamda = cp.array([1.0]) cp.testing.assert_allclose(thresh.hard_thresh(lamda, x), y)
def test_hard_thresh(self): x = np.array([-2, -1.5, -1, 0.5, 0, 0.5, 1, 1.5, 2]) y = np.array([-2, -1.5, 0, 0, 0, 0, 0, 1.5, 2]) npt.assert_allclose(thresh.hard_thresh(1, x), y)