(self)
| 37 | |
| 38 | @test_util.run_deprecated_v1 |
| 39 | def testTrace(self): |
| 40 | for dtype in [np.int32, np.float32, np.float64]: |
| 41 | for shape in [[2, 2], [2, 3], [3, 2], [2, 3, 2], [2, 2, 2, 3]]: |
| 42 | x = np.random.rand(np.prod(shape)).astype(dtype).reshape(shape) |
| 43 | self.compare(x) |
| 44 | |
| 45 | |
| 46 | if __name__ == "__main__": |