训练时报错

[heppytt]

InvalidArgumentError (see above for traceback): ConcatOp : Dimensions of inputs should match: shape[0] = [128,1024] vs. shape[1] = [640,1024]
[[Node: decoder/decoder/while/BasicDecoderStep/decoder/Attention_Wrapper/concat = ConcatV2[N=2, T=DT_FLOAT, Tidx=DT_INT32, _class=["loc:@decod...apeN/f_acc"], _device="/job:localhost/replica:0/task:0/device:CPU:0"](decoder/decoder/while/Switch_12:1, decoder/decoder/while/Switch_8:1, decoder/decoder/while/BasicDecoderStep/decoder/Attention_Wrapper/concat/axis)]]

[xuyl0104]

我遇到了同样的问题，请问你的问题解决了吗 @heppytt
估计是beam_search扩大维度的时候出错了，我没好到哪里出问题。能否帮我们解答一下？ @lc222 谢谢！