Я новичок в tf.keras функциональном API.Сейчас я строю модель, которая кодирует определенный диапазон слов в последовательности предложений по слоям LSTM.Например, ввод: [None, max_sent_len, word_embedding_dim], span: [2, 5] должен передать слово index[2,3,4] в слой LSTM и вывести последний скрытый слой.Есть ли в tf.keras какая-нибудь функция, которая может это легко построить?Или я должен использовать низкоуровневый тензорный поток для создания этого?