For an encoder decoder task, I want to get every first element in the whole batch to feed it into the decoder (first element of every document is always the  tag).
Let’s say I have this output tensor:
tensor([[   1, 1870, 1871,  ...,    0,    0,    0],
        [   1, 1159, 1160,  ...,    0,    0,    0],
        [   1, 1159, 1500,  ..., 1651, 1170, 1167],
        [   1, 1172, 1173,  ...,    0,    0,    0]])

How can I get:
tensor([[   1  ],
        [   1  ],
        [   1  ],
        [   1  ]])

I am following this tutorial and the part which doesn’t work for me is under “9. Seq2Seq (Encoder + Decoder) Code Implementation” in line 22.
Has anyone an idea? Thank you!