Python get_vq_codebook Examples

Programming Language: Python

Namespace/Package Name: tensor2tensor.layers.discretization

Method/Function: get_vq_codebook

Examples at hotexamples.com: 2

Python get_vq_codebook - 2 examples found. These are the top rated real world Python examples of tensor2tensor.layers.discretization.get_vq_codebook extracted from open source projects. You can rate examples to help us improve the quality of examples.

Example #1

Show file

File: discretization_test.py Project: qixiuai/tensor2tensor

 def testGetVQBottleneck(self):
   bottleneck_bits = 2
   bottleneck_size = 2**bottleneck_bits
   hidden_size = 3
   means, _, ema_count = discretization.get_vq_codebook(
       bottleneck_size, hidden_size)
   assign_op = means.assign(tf.zeros(shape=[bottleneck_size, hidden_size]))
   means_new, _, _ = discretization.get_vq_codebook(bottleneck_size,
                                                    hidden_size)
   with self.test_session() as sess:
     tf.global_variables_initializer().run()
     sess.run(assign_op)
     self.assertTrue(np.all(sess.run(means_new) == 0))
     self.assertTrue(np.all(sess.run(ema_count) == 0))

Example #2

Show file

 def testGetVQBottleneck(self):
   bottleneck_bits = 2
   bottleneck_size = 2**bottleneck_bits
   hidden_size = 3
   means, _, ema_count = discretization.get_vq_codebook(
       bottleneck_size, hidden_size)
   assign_op = means.assign(tf.zeros(shape=[bottleneck_size, hidden_size]))
   means_new, _, _ = discretization.get_vq_codebook(bottleneck_size,
                                                    hidden_size)
   with self.test_session() as sess:
     tf.global_variables_initializer().run()
     sess.run(assign_op)
     self.assertTrue(np.all(sess.run(means_new) == 0))
     self.assertTrue(np.all(sess.run(ema_count) == 0))