Python _flatten_beam_dim Beispiele

Beispiel #1

Datei: beam_search_test.py Projekt: cybermaster/reference

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

Beispiel #2

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

Beispiel #3

Datei: beam_search_test.py Projekt: zxfzqf/TransformerTranslationModel

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