def testBasic(self): for dtype in [np.float32, np.float64]: values = [0, 1, np.nan, np.inf, np.NINF] x = constant_op.constant(values, dtype=dtype) zeros = constant_op.constant(np.zeros((5, )), dtype=dtype) ones = constant_op.constant(np.ones((5, )), dtype=dtype) with test_util.use_gpu(): tf_result_zeros = math_ops.multiply_no_nan(x, zeros) self.assertAllEqual(tf_result_zeros, zeros) tf_result_ones = math_ops.multiply_no_nan(x, ones) self.assertAllEqual(tf_result_ones, x)
def testBasic(self): for dtype in [np.float32, np.float64]: values = [0, 1, np.nan, np.inf, np.NINF] x = constant_op.constant(values, dtype=dtype) zeros = constant_op.constant(np.zeros((5,)), dtype=dtype) ones = constant_op.constant(np.ones((5,)), dtype=dtype) with self.cached_session(use_gpu=True): tf_result_zeros = math_ops.multiply_no_nan(x, zeros).eval() self.assertAllEqual(tf_result_zeros, zeros) tf_result_ones = math_ops.multiply_no_nan(x, ones).eval() self.assertAllEqual(tf_result_ones, x)
def testBasic(self): for dtype in [np.float32, np.float64]: values = [0, 1, np.nan, np.inf, np.NINF] x = constant_op.constant(values, dtype=dtype) zeros = constant_op.constant(np.zeros((5,)), dtype=dtype) ones = constant_op.constant(np.ones((5,)), dtype=dtype) with test_util.use_gpu(): tf_result_zeros = math_ops.multiply_no_nan(x, zeros) self.assertAllEqual(tf_result_zeros, zeros) tf_result_ones = math_ops.multiply_no_nan(x, ones) self.assertAllEqual(tf_result_ones, x) # Normal floating point arithmetic if nonfinite values are in the # second argument. tf_result_reverseargs = math_ops.multiply_no_nan(zeros, x) self.assertAllEqual(zeros * x, tf_result_reverseargs)