From 0c3bae3842ade8a33af2a9d98c253b7aca2c0ee9 Mon Sep 17 00:00:00 2001 From: Edward Loper Date: Tue, 3 Sep 2019 09:16:45 -0700 Subject: [PATCH] Add shape test for ragged reduce_sum. PiperOrigin-RevId: 266944199 --- tensorflow/python/ops/ragged/ragged_reduce_op_test.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tensorflow/python/ops/ragged/ragged_reduce_op_test.py b/tensorflow/python/ops/ragged/ragged_reduce_op_test.py index db6c792b1e8..9331bc6e14f 100644 --- a/tensorflow/python/ops/ragged/ragged_reduce_op_test.py +++ b/tensorflow/python/ops/ragged/ragged_reduce_op_test.py @@ -321,6 +321,13 @@ class RaggedReduceOpsTest(test_util.TensorFlowTestCase, reduced = ragged_reduce_op(rt_input, axis) self.assertAllEqual(reduced, expected) + def testReduceKeepsInnerDimensionShape(self): + # Test for bug [b/139823356]. + rt = ragged_factory_ops.constant([[[[1, 1]]]], ragged_rank=2) + self.assertEqual(rt.shape.as_list(), [1, None, None, 2]) + reduced = ragged_math_ops.reduce_sum(rt, axis=2) + self.assertEqual(reduced.shape.as_list(), [1, None, 2]) + def assertEqualWithNan(self, actual, expected): """Like assertEqual, but NaN==NaN.""" self.assertTrue(