Jacky_Liu:

r_out2 = self.linear(r_out[:, -1, :])


@Jacky_Liu
r_out2 = self.linear(r_out[:, -1, :])
after this arent u loosing the original batch size of ,batch_sizenum of steps
So we will have target input size mismatch   when calculate the loss
our target size is still batch size * num of steps 1