batch_size и max_time в LSTM Tensorflow - PullRequest
       31

batch_size и max_time в LSTM Tensorflow

0 голосов
/ 11 февраля 2019

Справочная информация: Я пытаюсь смоделировать многослойный LSTM в тензорном потоке.Я использую обычную функцию для развертывания LSTM:

tf.nn.dynamic_rnn

Здесь я использую time_major = True, поэтому мои данные должны иметь формат [max_time, batch_size, depth].

В моем понимании max_time - временной шаг серии.Мой ввод [224], я вначале перехожу на слой FC, чтобы сделать его размером меток

Вопрос :

Я использую линию данных для полученияметки в пакете (32) вектора one_hot (длина = 70).Вектор длины

70 соответствует 1 временному шагу.

Итак, как я могу сделать ввод для [32, 32, 70]?В настоящее время у меня есть [32,70] (batch_size,num_classes).

Пожалуйста, исправьте мое понимание, если оно неверно.

Могу ли я просто передать [1,32,70], чтобы тензор потока самостоятельно обнаружил, что каждая партия имеет шаг времени 1?

...