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