Method compare

tensorflow/python/kernel_tests/trace_op_test.py:32–36

(self, x)

    x = np.random.seed(0)

  def compare(self, x):
    np_ans = np.trace(x, axis1=-2, axis2=-1)
    with self.cached_session(use_gpu=True):
      tf_ans = math_ops.trace(x).eval()
    self.assertAllClose(tf_ans, np_ans)

  @test_util.run_deprecated_v1
  def testTrace(self):