Add test for sequence_softmax.

del_some_in_makelist
yangyaming 7 years ago
parent 6cba821a60
commit 760d20de92

@ -187,6 +187,15 @@ class TestBook(unittest.TestCase):
x_t=x_t, hidden_t_prev=prev_hidden, cell_t_prev=prev_cell))
print(str(program))
def test_sequence_softmax(self):
program = Program()
with program_guard(program):
seq_data = layers.data(
name='seq_data', shape=[10, 10], dtype='float32', lod_level=1)
seq = layers.fc(input=seq_data, size=20)
self.assertIsNotNone(layers.sequence_softmax(x=seq))
print(str(program))
if __name__ == '__main__':
unittest.main()

Loading…
Cancel
Save