TypeError：__init__() 為參數“軸”獲取了多個值

Question

我想在下面的代碼中使用注意機制：

attention = Dot([decoder_outputs, encoder_outputs], axes=[2, 1])
attention = Activation('softmax')(attention)

context = Dot([attention, encoder_outputs], axes=[2,1])
decoder_combined_context = concatenate([context, decoder_outputs])

但我有這個錯誤信息，我不知道如何解決它

Traceback (most recent call last):

    attention = Dot([decoder_outputs, encoder_outputs], axes=[2, 1])

TypeError: __init__() got multiple values for argument 'axes'

我正在使用 keras 版本 2.3.1 和 tensorflow 版本 2.1.0

Answer 1

Dot的第一個參數稱為axes ：

tf.keras.layers.Dot(axes, normalize=False, **kwargs)

因此，您需要決定要axes傳遞什么值： [decoder_outputs, encoder_outputs]或[2, 1] 。

Answer 2

您沒有正確調用圖層，它應該是這樣的：

attention = Dot(axes=[2, 1])([decoder_outputs, encoder_outputs])

您對Dot的第二次調用有同樣的問題，它遵循相同的模式