should be:
        src (encoder input)              tgt (decoder input)
<sos> tok0 tok1 tok2 <eos> <pad>  ->  <sos> tok0 tok1 tok2 <eos>
                                       output (decoder output)
                                      tok0 tok1 tok2 <eos> <pad>