当前位置:网站首页>【科研-学习-pytorch】5-boardcasting、合并分割
【科研-学习-pytorch】5-boardcasting、合并分割
2022-08-09 00:25:00 【helloworld_Fly】
boardcasting(自动推广)
- 首维插入1
- 将维度为1扩展为相同维度
- 特征尺寸匹配
- 扩展为相同尺寸后相加
boardcasting可理解为:unsqueeze + expand调整维度一致
boardcasting是由需求而来,对于不同尺寸的计算,需要手动多次处理设置尺寸一致,使用boardcasting可以自动匹配尺寸且不需要额外占用内存,准确方便且高效;
boardcasting使用范围
- 从最后一维开始匹配;
- 若高维没有则先扩展1维,后匹配一致;
- 低维需要可理解使用,如维度一致或者只有1;
- 无法使用情况-input [4, 3, 32, 32]
- [2, 3, 32, 32]
- [8]
- [1, 2, 32, 32]
应用
input [4, 3, 32, 32]
- [32, 32] h,w方向整体平移
- [3, 1, 1] rgb通道调整
- [1, 1, 1, 1] 所有像素加
合并
cat
cat前提是:两特征维度一致,要拼接维度可以不一致;
stack
创造新的维度,叠加;
a = torch.rand(3, 24)
b = torch.rand(3, 24)
c = torch.stack(a, b, dim=0)
c.shape = [2, 3, 24]
拼接维度必须一致,才能创建新维度拼接;
分割
split
split([2, 2], dim=0) 和 split(1, dim=0)分别拆分对应维度的长度拆分;
长度拆分和数量拆分;
chunck
按数量拆分,chunk按数值均匀拆分!
边栏推荐
猜你喜欢
随机推荐
一道dp的三次优化(时间+空间)
非科班毕业生,五面阿里:四轮技术面 +HR 一面已拿 offer
牛客练习赛87
求所有子串的最大值之和
Dart高级(一)——泛型与Json To Bean
Mysql Workbench uses .sql file to import data into database
自考成绩总结
There is quality when someone is in charge: to a generation lost in the market place
[Deep Learning] TensorFlow Learning Road 2: Introduction to ANN and TensorFlow Implementation
京东三面惨遭被虐,关于 redis,高并发,分布式,问懵了
Data Agencies - Huffman Trees
遮罩纹理-MaskTexture
【科研-学习-pytorch】3-分类问题
牛客练习赛88 补题
【学习-目标检测】目标检测之——YOLO v3
一名双非程序媛面试蚂蚁、美团、携程等大厂拿 offer 分享面试过程
整流八--电网不平衡状态下三相PWM整流器的控制策略
【科研-学习-pytorch】1-框架特性和常见问题类型
Mysql 根据一个表数据更新另外一个表
逐片元-兰伯特光照模型