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))