def test_basic(self):
    state_dim = 8
    model = ExampleSequenceModel(state_dim)
    input = model.input_prototype()
    output = model(input)
    self.assertEqual((1, 1), output.value.shape)
def test_basic(self):
    state_dim = 8
    model = ExampleSequenceModel(state_dim)
    input = model.input_prototype()
    output = model(input)
    self.assertEqual((1, 1), output.value.shape)