6
Привет Я использую следующую функцию для ячейки lstm rnn.Тензорный поток - LSTM - Объект «Tensor» не повторяется
def LSTM_RNN(_X, _istate, _weights, _biases):
# Function returns a tensorflow LSTM (RNN) artificial neural network from given parameters.
# Note, some code of this notebook is inspired from an slightly different
# RNN architecture used on another dataset:
# https://tensorhub.com/aymericdamien/tensorflow-rnn
# (NOTE: This step could be greatly optimised by shaping the dataset once
# input shape: (batch_size, n_steps, n_input)
_X = tf.transpose(_X, [1, 0, 2]) # permute n_steps and batch_size
# Reshape to prepare input to hidden activation
_X = tf.reshape(_X, [-1, n_input]) # (n_steps*batch_size, n_input)
# Linear activation
_X = tf.matmul(_X, _weights['hidden']) + _biases['hidden']
# Define a lstm cell with tensorflow
lstm_cell = rnn_cell.BasicLSTMCell(n_hidden, forget_bias=1.0)
# Split data because rnn cell needs a list of inputs for the RNN inner loop
_X = tf.split(0, n_steps, _X) # n_steps * (batch_size, n_hidden)
# Get lstm cell output
outputs, states = rnn.rnn(lstm_cell, _X, initial_state=_istate)
# Linear activation
# Get inner loop last output
return tf.matmul(outputs[-1], _weights['out']) + _biases['out']
Выход функции хранится в переменной pred.
pred = LSTM_RNN(x, istate, weights, biases)
Но его показ следующее сообщение об ошибке. (В котором говорится, что тензор объект не итерация.)
Вот ссылка ERROR изображения - http://imgur.com/a/NhSFK
Пожалуйста, помогите мне с этим, и я прошу прощения, если этот вопрос кажется глупым, как я довольно новым для LSTM и тензора библиотека потока.
Спасибо.
его команда изменения, проверьте, что http://stackoverflow.com/questions/33884978/build-a-graph-that-works-with-variable-batch-size-using-tensorflow –
, но Команда reshape необходима для преобразования в требуемую форму. –
уверен, попробуйте с tf.reshape (x, tf.pack (n_input, -1])) –