Can we have the pytorch "return_state_sequence" = True in tensorflow version like in return_seq=True in tensorflow / keras original code ?