code: def testOneHotUint8WithLargeArray(self):
      with self.cached_session(use_gpu=False) as sess:
          matrix = np.random.rand(256) * 10
          tensor = constant_op.constant(
              matrix, dtypes.uint8, shape=matrix.shape
          )
          tensor_one_hot = array_ops.one_hot(tensor, depth=10, axis=0)
          self.assertEqual(sess.run(tensor_one_hot).shape, (10, 256))
  
imports: from absl.testing import parameterized
  import numpy
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.ops import array_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 20 for calling the OneHotTest obj's underlying member method code with example usage: obj.testOneHotUint8WithLargeArray() without return types