如何通过tf.keras.layers.LSTM在文本序列中编码跨度

Question

我是tf.keras功能API的新手。我现在正在建立一个模型，该模型通过LSTM层对句子序列中的某些单词跨度进行编码。例如，输入为：[None, max_sent_len, word_embedding_dim]，跨度：[2, 5]应将index[2,3,4]的单词传递到LSTM层并输出最后一个隐藏层。 tf.keras中是否有任何函数可以轻松构建此函数？还是应该使用低级张量流来构建它？