小编use*_*490的帖子

Tensorflow:LSTM 中的形状错误,“lstm”层有多个入站节点,具有不同的输出形状

我收到有关 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)

python lstm tensorflow recurrent-neural-network

4
推荐指数
1
解决办法
2309
查看次数