(七)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]
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]