我收到有关 lstm 层输出形状的奇怪错误。我尝试了几件事,但不确定我在哪里做错了。
这个问题来自courser的深度学习专业
''' def music_inference_model(LSTM_cell, densor, Ty=100):
n_values = densor.units
n_a = LSTM_cell.units
x0 = Input(shape=(1, n_values))
a0 = Input(shape=(n_a,), name='a0')
c0 = Input(shape=(n_a,), name='c0')
a = a0
c = c0
x = x0
outputs = []
for t in range(Ty):
a, _, c = LSTM_cell(x, initial_state=[a, c])
out = densor(a)
outputs.append(out)
x = tf.math.argmax(out)
x = tf.one_hot(x, depth=n_values)
x = RepeatVector(1)(x)
inference_model = Model([x0,a0,c0],outputs)
return inference_model
inference_model = music_inference_model(LSTM_cell, densor, Ty = 50)
inference_summary = summary(inference_model)
comparator(inference_summary, …Run Code Online (Sandbox Code Playgroud)