不同shape直接加减,系统会自动做broadcasting操作
先右对齐(小维度对齐)
比如:Feature maps: [4,32,14,14]
Bias:[32,1,1]
=>][1,32,1,1]
=>[4,32,14,14] 做到与Feature maps的shape相同,才能进行相加
广播扩展的时候只是做这样的操作,并不实质拷贝数据,以节省内存空间
可广播的条件:右对齐,有dim的size必须是1(或者size相同),再扩展到相同size;没有dim,先插入dim,size=1,再扩展到相同size。否则不符合广播操作条件。
标签:dim,tensor,32,广播,pytorch,broadcasting,对齐,size,14 From: https://blog.csdn.net/vmt/article/details/140925662