LSTMCell初始化问题(get_initial_state)TF2.0

时间:2019-12-19 06:05:09

标签: tensorflow2.0

当我使用get_initial_state初始化LSTMCell时,出现以下错误。 您能建议我如何初始化LSTMCell单元吗?

lstm_cell = tf.keras.layers.LSTM(单位= 128) lstm_cell = tf.nn.RNNCellDropoutWrapper(lstm_cell,output_keep_prob = self.dropout_keep_prob) self._initial_state = lstm_cell.get_initial_state(128,tf.float32)

ValueError:维度0的切片索引0超出范围。对于输入形状为[0],[1],[1],[1]且计算输入张量为:输入[1] = <0>,输入[2] = <1>的strided_slice(op:StridedSlice),输入[3] = <1>。

0 个答案:

没有答案