math_ops_test.py 文件源码

def testRounding(self):
    try:
      x = [0.49, 0.7, -0.3, -0.8]
      for dtype in [np.float32, np.double]:
        x_np = np.array(x, dtype=dtype)
        for use_gpu in [True, False]:
          with self.test_session(use_gpu=use_gpu):
            x_tf = constant_op.constant(x_np, shape=x_np.shape)
            y_tf = math_ops.round(x_tf)
            y_tf_np = y_tf.eval()
            y_np = np.round(x_np)
            self.assertAllClose(y_tf_np, y_np, atol=1e-2)
    except:
      import sys, pdb, traceback
      type, value, tb = sys.exc_info()
      traceback.print_exc()
      pdb.post_mortem(tb)