stack使用报错
Created by: Angus07
在运行阶段: 注意到这一行,好像是这里错误,但是看输入的形状应该是正确的: I1120 15:16:47.257516 65396 operator.cc:152] CUDAPlace(0) Op(stack), inputs:{X[reshape2_52.tmp_0:int3200, 1600, squeeze_0.tmp_0:int64_t3200, 1600]}, outputs:{Y[stack_2.tmp_0:0]}.
相关代码: new_src_ids = layers.stack([dec, src_ids], axis=2) new_src_ids.stop_gradient= True 其中dec和src_ids都是[3200, 1600]