(self, dev)
| 585 | self._MeanSquareError_helper(gpu_dev) |
| 586 | |
| 587 | def _Abs_helper(self, dev): |
| 588 | X = np.array([0.8, -1.2, 3.3, -3.6, -0.5, |
| 589 | 0.5]).reshape(3, 2).astype(np.float32) |
| 590 | XT = np.array([0.8, 1.2, 3.3, 3.6, 0.5, |
| 591 | 0.5]).reshape(3, 2).astype(np.float32) |
| 592 | x = tensor.from_numpy(X) |
| 593 | x.to_device(dev) |
| 594 | |
| 595 | result = autograd.abs(x) |
| 596 | dx = result.creator.backward(x.data) |
| 597 | |
| 598 | np.testing.assert_array_almost_equal(tensor.to_numpy(result), XT) |
| 599 | self.check_shape(dx.shape(), (3, 2)) |
| 600 | |
| 601 | def test_Abs_cpu(self): |
| 602 | self._Abs_helper(cpu_dev) |
no test coverage detected