class_docstr: Tests for discretization layers.
code: def testVQDiscreteUnbottlenck(self):
      x = tf.constant([[1, 0, 0, 0], [0, 0, 1, 0]], dtype=tf.int32)
      x_means = discretization.vq_discrete_unbottleneck(
          x, hidden_size=3
      )
      self.evaluate(tf.global_variables_initializer())
      x_means_eval = self.evaluate(x_means)
      self.assertEqual(np.shape(x_means_eval), (2, 3))
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import numpy
  from tensor2tensor.layers import discretization
  from tensor2tensor.utils import test_utils
  import tensorflow.compat.v1
label_desc: Use image node_id 13 for calling the DiscretizationTest obj's underlying member method code with example usage: obj.testVQDiscreteUnbottlenck() without return types