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]
本文标题：(七)Broadcast自动扩展
本文链接：https://www.haomeiwen.com/subject/gxsdactx.html
网友评论