add的广播机制输出shape问题

[ljbulesky]

当add两个输入是[197, 768] 和 [1, 197, 768]的时候（N,C,H,W），使用广播机制，输出是【197， 768， 768】 而我需要的正确结果是[1, 197, 768]

请教一下，是不是二维与三维的广播出现了问题， 能否将 B2 = B.reshape(1, B.w, B.h, opt.workspace_allocator); 修改为 B2 = B.reshape(B.w, B.h, 1, opt.workspace_allocator);

@nihui 非常感谢

[nihui]

https://github.com/Tencent/ncnn/wiki/binaryop-broadcasting

支持的广播规则并没有这种情况？

[ljbulesky]

感谢解答，意思是不支持这种情况吗？但是这应该属于常见情况吧，我修改成B2 = B.reshape(B.w, B.h, 1, opt.workspace_allocator);输出shape就对了。 为啥不按这种做个判断啊？ if (outdims == 3 && A.dims == 1) { if (A.w == B.c) A2 = A.reshape(1, 1, A.w); else // if (A.w == B.w) A2 = A.reshape(A.w, 1, 1); } @nihui