示例#1
0
 def testGetTimingSignal(self):
     length = 7
     num_timescales = 10
     a = common_layers.get_timing_signal(length,
                                         num_timescales=num_timescales)
     res = self.evaluate(a)
     self.assertEqual(res.shape, (length, 2 * num_timescales))
 def testGetTimingSignal(self):
   length = 7
   num_timescales = 10
   with self.test_session() as session:
     a = common_layers.get_timing_signal(length, num_timescales=num_timescales)
     session.run(tf.global_variables_initializer())
     res = session.run(a)
   self.assertEqual(res.shape, (length, 2 * num_timescales))
 def testGetTimingSignal(self):
   length = 7
   num_timescales = 10
   a = common_layers.get_timing_signal(length, num_timescales=num_timescales)
   res = self.evaluate(a)
   self.assertEqual(res.shape, (length, 2 * num_timescales))