[Transformer] Be more flexible by passing directly the whole context

Actually there is no good rease to pass only target. 

Let pass the context to be more extensible.

What do you think ?