从零开始编写深度学习库（五）Eigen Tensor学习笔记2.0
2024-09-01 17:32:33
1、extract_image_patches函数的使用：
假设Eigen::Tensor形状为(3,8,8,9),现在要对第二维、第三维根据size大小为(2,2),stride=(2,2)，那么如果tensor类型是rowmajor类型，那么经过extract_image_patches后的数据就是(3,4*4,2,2,9)的5维数组，如果tensor类型是colmajor类型，那么得到的数据就是(3,2,2,4*4,9)的五维数组。
Eigen::array<int, 2> reduction_dims{2,3};
Tensor5xf patches = bottom.extract_image_patches(m_hksize, m_wksize, m_hstride, m_wstride, 1,1, Eigen::PADDING_VALID);
//std::cout << "post_reduce_dims"<< post_reduce_dims << std::endl;
Tensor3xf pooling.device() = patches.maximum(reduction_dims);
top=pooling.reshape(post_reduce_dims);
另外需要注意的是如果size大小为(3,2)，那么如果是rowmajor那么数组的形状将是(3,*,2,3,9);如果是colmajor那么数组的形状将是(3,*,3,2,9)
2、降维统计
Eigen::array<int, 2> reduction_dims{2,3};
patches.mean(reduction_dims);//求取第2、3维度的平均值
patches.maximum(reduction_dims);//求取第2、3维度的最大值
3、数据存储方式交换：swap_layout()
比如如果Tensor变量是 colmajor，可以调用该函数，返回的新tensor就是rowmajor格式。
4、
最新文章
- UiAutomator--UiAutomatorHelper快速调试
- C#中Object和Json之间的转换
- node log4js包
- onsubmit="return false;"报错
- 转 Learning To Rank之LambdaMART的前世今生
- LNA和PA
- C# toString()转换详细（转）
- SQL注入测试用例
- 转载：div和flash层级关系问题
- 我的开发框架(WinForm)4
- 初探performance.timing API
- 算法模板——Dinic最小费用最大流
- BezierDemo开源项目的学习
- meta的用法
- HEX SDUT 3896 17年山东省赛D题
- Aes CBC加密
- MySQL拓展操作
- 机械革命 x7ti-s 1周年使用报告
- 命名和目录接口 JNDI-The Java Naming and Directory Interface
- JSP怎么将表单提交到对应的servlet