class_docstr: Tests for discretization layers.
code: def testSliceHiddenZeros(self):
      hidden_size = 60
      block_dim = 20
      num_blocks = 3
      x = tf.zeros(shape=[1, 1, hidden_size], dtype=tf.float32)
      x_sliced = discretization.slice_hidden(x, hidden_size, num_blocks)
      x_sliced_eval = self.evaluate(x_sliced)
      self.assertEqual(
          np.shape(x_sliced_eval), (1, 1, num_blocks, block_dim)
      )
      self.assertTrue(np.all(x_sliced_eval == 0))
  
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 7 for calling the DiscretizationTest obj's underlying member method code with example usage: obj.testSliceHiddenZeros() without return types