Python _unflatten_beam_dim Beispiele

Beispiel #1

Datei: beam_search_test.py Projekt: cybermaster/reference

 def test_unflatten_beam_dim(self):
   x = tf.ones([28, 2, 5])
   x = beam_search._unflatten_beam_dim(x, 7, 4)
   with self.test_session() as sess:
     shape = sess.run(tf.shape(x))
   self.assertAllEqual([7, 4, 2, 5], shape)

Beispiel #2

 def test_unflatten_beam_dim(self):
   x = tf.ones([28, 2, 5])
   x = beam_search._unflatten_beam_dim(x, 7, 4)
   with self.test_session() as sess:
     shape = sess.run(tf.shape(input=x))
   self.assertAllEqual([7, 4, 2, 5], shape)

Beispiel #3

Datei: beam_search_test.py Projekt: zxfzqf/TransformerTranslationModel

 def test_unflatten_beam_dim(self):
     x = nd.ones([28, 2, 5], ctx=ctx)
     x = beam_search._unflatten_beam_dim(x, 7, 4)
     self.assertEqual((7, 4, 2, 5), x.shape)