设置 Keras 点轴

问题描述

我正在尝试使用 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)

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)