pytorch & numpy广播法则
2024-10-08 14:33:56
广播法则
- 所有数组向维度最高的数组看齐,若维度不足则在最前面的维度用1补齐
- 扩展维度后,所有数组在某一维度相同或者长度为1,否则不能计算
- 当可以计算时,将长度为1的维度扩展为另一数组相应维度的长度
a = torch.ones(3, 2)
b = torch.zeros(2,3,1)
a + b
# a : (3, 2)-->(1, 3, 2)
# a : (1, 3, 2)-->(2, 3, 2)
# b : (2, 3, 1)-->(2, 3, 2)
# a + b : (2, 3, 2)
手工实现广播(建议,较为直观):
a.view(1, 3, 2).expand(2, 3, 2)
b.expand(2, 3, 2)
# repeat和expand功能类似,但是repeat会把数据复制多份,会占用额外空间
最新文章
- ng-class结合三目运算
- 【BZOJ-3507】通配符匹配 DP + Hash
- How to load a local .CSS file &; JavaScript resources using iPhone UIWebView Class
- Android应用安全之Android APP通用型拒绝服务漏洞
- Fresco 源码分析(三) Fresco服务端处理(2) Producer具体实现的内容
- java面试---summay
- Eclipse自动补全功能和自动生成作者、日期注释等功能设置
- pg 资料大全1
- Java基础——异常处理
- python中变量
- HDU4720+三角形外接圆
- Asp.Net MVC 模型(使用Entity Framework创建模型类) - Part.1
- c#读取文件
- 我永远的 dell 15r
- windows phone7开发环境配置错误
- [HAOI2015]数字串拆分
- SpriteBuilder复杂CCB在App场景加载时报错排查
- Codeforces551 C. GukiZ hates Boxes
- 向redis中添加删除list列表
- 接口自动化测试 (三)request.post