Both have shape (batch_size, sequence_length).