1:Mul维度不匹配问题
通过onnx找到对应输出输入的维度形状:
bottom[0]->对应294,维度是2x16x1x1
bottom[1]->对应284,维度是2x16x28x28
解决思路:利用flatten和scale层实现Mul
1:交换bottom[0]和bottom[1]:因为scale层机制是广播第二个输入
bottom[0]->对应284,维度是2x16x28x28
bottom[1]->对应294,维度是2x16x1x1
2:利用flatten压缩bottom[1]
bottom[1]->对应294,维度变成2x16
3:利用scale层实现广播乘法
bottom[0]->对应284,维度是2x16x28x28
bottom[1]->对应294,维度是2x16
具体操作:
修