Method random_func

tensorflow/python/data/kernel_tests/map_test.py:874–877

(x)

      self.assertFalse(found_warning)

      def random_func(x):
        x = math_ops.add(x, 1)
        random_ops.random_shuffle([x, math_ops.square(x)])
        return x

      with warnings.catch_warnings(record=True) as w:
        _ = dataset_ops.Dataset.range(10).map(random_func)