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)