def __call__(self, inputs): lstm_inputs, time = inputs cell_mask = self.cell_mask(time) outputs = self.cell_masked_lstm( concatenate([lstm_inputs, cell_mask], axis=2) ) return outputs