Solved by Pranavan_Theivendira in post #2 
              
              
                Thanks for the people who read this topic. I found the solution. In keras, return_sequences=False actually returns the final output in the sequence dimension. Hence, it can be sliced from the GRU output of pytorch.