def test_LeakyRelu1D(self): npr = self.np_a.copy() npr[self.np_a < 0] = npr[self.np_a < 0] * self.alpha dcr = dc.leakyrelu(self.dc_a, self.alpha) np.testing.assert_allclose(npr, np.array(dcr.data()).astype(np.float32), rtol=1e-3, atol=1e-3)
def test_LeakyRelu4D(self): np_a = np.reshape(self.np_a, (2, 2, 2, 3)) dc_a = dc.reshape(self.dc_a, (2, 2, 2, 3)) npr = self.np_a.copy() npr[self.np_a < 0] = npr[self.np_a < 0] * self.alpha dcr = dc.leakyrelu(self.dc_a, self.alpha) np.testing.assert_allclose(npr.flatten(), np.array(dcr.data()).astype(np.float32), rtol=1e-3, atol=1e-3)