engine.py 文件源码

def __init__(self, state_sync=False, decode=False, output_length=None, return_states=False, readout=False, readout_activation='linear', teacher_force=False, state_initializer=None, **kwargs):
        self.state_sync = state_sync
        self.cells = []
        if decode and output_length is None:
            raise Exception('output_length should be specified for decoder')
        self.decode = decode
        self.output_length = output_length
        if decode:
            if output_length is None:
                raise Exception('output_length should be specified for decoder')
            kwargs['return_sequences'] = True
        self.return_states = return_states
        super(RecurrentModel, self).__init__(**kwargs)
        self.readout = readout
        self.readout_activation = activations.get(readout_activation)
        self.teacher_force = teacher_force
        self._optional_input_placeholders = {}
        if state_initializer:
            if type(state_initializer) in [list, tuple]:
                state_initializer = [initializers.get(init) if init else initializers.get('zeros') for init in state_initializer]
            else:
                state_initializer = initializers.get(state_initializer)
        self._state_initializer = state_initializer