网站首页
编程语言
数据库
系统相关
其他分享
编程问答
broadcasting
2024-08-05
pytorch学习笔记5 tensor 广播broadcasting
不同shape直接加减,系统会自动做broadcasting操作先右对齐(小维度对齐)比如:Featuremaps:[4,32,14,14]Bias:[32,1,1]=>][1,32,1,1]=>[4,32,14,14]做到与Featuremaps的shape相同,才能进行相加广播扩展的时候只是做这样的操作,并不实质拷贝数据,以节省内存空间可广播的条件