突唯阿 领先的响应式网站建设平台,莞城微网站建设,深圳公司网站开发,台州市建设局网站broadcast#xff1a; 1、能够进行维度扩展#xff0c;是自动的2、在扩展的时候不需要拷贝数据要点#xff1a; - 从最小的维度开始匹配#xff0c;如果前面没有维度了#xff0c;在前面插入一个新的维度。- 插入的新维度size是1#xff0c;再将其扩展为与目标相同大小si…broadcast 1、能够进行维度扩展是自动的2、在扩展的时候不需要拷贝数据要点 - 从最小的维度开始匹配如果前面没有维度了在前面插入一个新的维度。- 插入的新维度size是1再将其扩展为与目标相同大小size。一个实际案例
[class, students, scores]
[4, 32, 8 ]
# 有4个班每个班32个学生每个学生都有8门课程由于今年期末考试难度过大校长决定给每个同学各科成绩加5分
5.0 - [1]
如果手动操作[1].unsqueeze(0).unsqueeze(0).expand(4, 32, 8)
# 一个浮点数占用4byte如果扩张成[4,32,8]的shape需要占用4096bytebroadcast能大大节约内存注在最小维度匹配才能继续扩展