Conditioned Recurrent

gistfile1.py

class ConditionedRecurrent(BaseRecurrent):

    def __init__(self, wrapped, **kwargs):
        super(ConditionedRecurrent, self).__init__(**kwargs)
        self.wrapped = wrapped
        self.children=[wrapped]

    def get_dim(self, name):
        if name == 'context':
            return self.wrapped.get_dim('inputs')
        return self.wrapped.get_dim(name)

    @application(contexts=['context'])
    def apply(self, **kwargs):
        context = kwargs['context']
        kwargs['inputs'] += context
        for context in ConditionedRecurrent.apply.contexts:
            kwargs.pop(context)
        return self.wrapped.apply(**kwargs)

    @apply.delegate
    def apply_delegate(self):
        return self.wrapped.apply