def testConvLSTM(self):
x = np.random.rand(5, 7, 11, 13)
y = common_layers.conv_lstm(tf.constant(x, dtype=tf.float32), (1, 3),
13)
self.evaluate(tf.global_variables_initializer())
res = self.evaluate(y)
self.assertEqual(res.shape, (5, 7, 11, 13))
def testConvLSTM(self):
x = np.random.rand(5, 7, 11, 13)
with self.test_session() as session:
y = common_layers.conv_lstm(tf.constant(x, dtype=tf.float32), (1, 3), 13)
session.run(tf.global_variables_initializer())
res = session.run(y)
self.assertEqual(res.shape, (5, 7, 11, 13))
def testConvLSTM(self): x = np.random.rand(5, 7, 11, 13) y = common_layers.conv_lstm(tf.constant(x, dtype=tf.float32), (1, 3), 13) self.evaluate(tf.global_variables_initializer()) res = self.evaluate(y) self.assertEqual(res.shape, (5, 7, 11, 13))
