Hi there,

I have a tensor with shape (N x T X D), representing embeddings of N sentences, and a tensor with shape (N x 2), representing the start and end indices of entities in the N sentences.

Is there a efficient way to slice out the embeddings of the entities and return a new tensor (N x L x D) where L is the length of entities?

Thanks