python - 设置 Keras 点轴

Question

我正在尝试使用 Keras Dot，但您能解释一下为什么在此示例代码中 Dot 的轴值为 2 吗？

w_inputs = Input(shape=(1, ), dtype='int32')
word_embedding = Embedding(vocab_size, embed_size)(w_inputs)

c_inputs = Input(shape=(1, ), dtype='int32')
context_embedding  = Embedding(vocab_size, embed_size)(c_inputs)

dot_product = Dot(axes=2)([word_embedding, context_embedding])
dot_product = Reshape((1,), input_shape=(1, 1))(dot_product)

标签： pythontensorflowkeraskeras-layerdot