(七)Broadcast自动扩展

2020-02-12  本文已影响0人  zelda2333

Broadcasting

具体实施方法

最后1维要匹配,将其他维度化为1
将1扩展为同样维度
A[4,32,14,14]
B[32,1,1]-->[1,32,1,1]-->[4,32,14,14]

A[4,32,14,14]
B[14,14]-->[1,1,14,14]-->[4,32,14,14]

上一篇 下一篇

猜你喜欢

热点阅读