def forward(self, s, **kwargs):
    's -> V(s)'
    (logits, h) = self.preprocess(s, state=kwargs.get('state', None))
    logits = self.last(logits)
    return logits
s -> V(s)
def forward(self, s, **kwargs):
    
    (logits, h) = self.preprocess(s, state=kwargs.get('state', None))
    logits = self.last(logits)
    return logits