diff --git a/PointerLSTM.py b/PointerLSTM.py index 4e16c78..04a9344 100644 --- a/PointerLSTM.py +++ b/PointerLSTM.py @@ -69,7 +69,7 @@ def build(self, input_shape): def call(self, x, training=None, mask=None, states=None): """ - :param Tensor x: Should be the output of the decoder + :param Tensor x: Should be the output of the encoder :param Tensor states: last state of the decoder :param Tensor mask: The mask to apply :return: Pointers probabilities