运行训练脚本时报如下错,是什么原因?
收藏
运行训练脚本时报错:
paddle.fluid.core.EnforceNotMet: enforce x_dims[i + axis] == y_dims[i] tailed 1!=32
Broadcast dimension mismatch. at [/paddle/paddle/fluid/operators/elementwise_op_function.h:59]
0
收藏
请登录后评论
应该是输入的类别和要输出的类别不相同吧。
有解决吗?
应该是张量的广播机制错误,建议检查一下数据,顺便补充一下广播机制的原则:
1. 维度相同,其中有一个轴为1
例如a1.shape=(4, 3),a2.shape=(4, 1)
2. 后置维度相同
例如a1.shape=(2, 3, 4),a2.shape=(3, 4),这里只用扩展一个维度;
a3.shape=(2, 3, 4, 5),a4.shape=(4, 5),这里需要扩展两个维度