code: def test2DInput(self):
      op = math_ops._bucketize(
          constant_op.constant([[-5, 0, 2, 3, 5], [8, 10, 11, 12, 0]]),
          boundaries=[0, 3, 8, 11],
      )
      expected_out = [[0, 1, 1, 2, 2], [3, 3, 4, 4, 1]]
      with self.session():
          self.assertAllEqual(expected_out, self.evaluate(op))
  
file_path: tensorflow-upstream/tensorflow/python/kernel_tests/math_ops/bucketize_op_test.py
imports: import numpy
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors_impl
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 4 for calling the BucketizationOpTest obj's underlying member method code with example usage: obj.test2DInput() without return types