Hi, Thanks Jaya !
For the Decoder and Encoder , i also thought this initially
But going forward we would have 2 hidden states for each sentence [as there are two sources per sentence[beam size=2]  each giving a new hidden state . Which dimension would pytorch put the 4 hidden states into?
I would need to feed four hidden states [2 sentences;2 beam size] which axis’s dimension should be increased?