def testEmptySplits(self):
     # Note: the splits for an empty ragged tensor contains a single zero.
     segment_ids = ragged.row_splits_to_segment_ids([0])
     self.assertAllEqual(segment_ids, [])
 def testDocStringExample(self):
     splits = [0, 3, 3, 5, 6, 9]
     expected = [0, 0, 0, 2, 2, 3, 4, 4, 4]
     segment_ids = ragged.row_splits_to_segment_ids(splits)
     self.assertAllEqual(segment_ids, expected)
Example #3
0
 def testEmptySplits(self):
     # Note: the splits for an empty ragged tensor contains a single zero.
     segment_ids = ragged.row_splits_to_segment_ids([0])
     with self.test_session():
         self.assertEqual(segment_ids.eval().tolist(), [])
Example #4
0
 def testDocStringExample(self):
     splits = [0, 3, 3, 5, 6, 9]
     expected = [0, 0, 0, 2, 2, 3, 4, 4, 4]
     segment_ids = ragged.row_splits_to_segment_ids(splits)
     with self.test_session():
         self.assertEqual(segment_ids.eval().tolist(), expected)
 def testEmptySplits(self):
   # Note: the splits for an empty ragged tensor contains a single zero.
   segment_ids = ragged.row_splits_to_segment_ids([0])
   with self.test_session():
     self.assertEqual(segment_ids.eval().tolist(), [])
 def testDocStringExample(self):
   splits = [0, 3, 3, 5, 6, 9]
   expected = [0, 0, 0, 2, 2, 3, 4, 4, 4]
   segment_ids = ragged.row_splits_to_segment_ids(splits)
   with self.test_session():
     self.assertEqual(segment_ids.eval().tolist(), expected)