a = K.permute_dimensions(a, (0, 3, 2, 1))

[monkeyshichi]

ValueError: Dimension must be 5 but is 4 for 'attention_1/transpose_7' (op: 'Transpose') with input shapes: [?,8,?,8,?], [4].

请问这个怎么解决 run example
https://kexue.fm/archives/4765
代码测试
在Keras上对IMDB进行简单的测试

[chaoswjc]

个人经验：
将a = K.batch_dot(qw, kw, [3, 3]) / self.key_size**0.5
改为a = tf.einsum('bjhd,bkhd->bhjk', qw, kw) / self.key_size**0.5

将o = K.batch_dot(a, vw, [3, 2])
改为o = tf.einsum('bhjk,bkhd->bjhd', a, vw)
主要是keras版本变动导致广播方式不一样了

[Yumeka999]

个人经验： 将a = K.batch_dot(qw, kw, [3, 3]) / self.key_size**0.5 改为a = tf.einsum('bjhd,bkhd->bhjk', qw, kw) / self.key_size**0.5

将o = K.batch_dot(a, vw, [3, 2]) 改为o = tf.einsum('bhjk,bkhd->bjhd', a, vw) 主要是keras版本变动导致广播方式不一样了

有用