
算子等效
在AI编译器中,静态图转化为动态中的阶段和深度学习框架前端转relay阶段,涉及到算子不支持的情况,当然也考虑到底层编译器不支持,我们可以利用现有算子去等效替代相应算子去节约开发时间。
胡乱儿起个名
工作记录.......
展开
-
转置卷积的实现过程
这里指的反卷积,也叫转置卷积,它并不是正向卷积的完全逆过程,反卷积是一种特殊的正向卷积,先按照一定的比例通过补0来扩大输入图像的尺寸,接着旋转卷积核,再进行正向卷积。输入特征图变换:对输入特征图进行变换,包括插值操作和填充操作。插值操作:当步长(stride)大于1时,需要在输入特征图的相邻值之间插入(stride-1)行和列0。这一步是为了在上采样过程中保持特征图的尺寸变化。填充操作:为了得到正确的输出尺寸,还需要在特征图的边缘进行填充。填充的像素数量与卷积核大小、步长和原始填充量有关。卷积运算。原创 2024-12-02 20:10:21 · 668 阅读 · 0 评论 -
卷积代替torch.diagonal()提取对角线元素
【代码】卷积代替torch.diagonal()提取对角线元素。原创 2024-11-13 11:10:47 · 717 阅读 · 0 评论 -
卷积和add、sub、mean等的转换
1的卷积,那么结果的大小为500。原创 2024-09-19 15:10:12 · 566 阅读 · 0 评论 -
深度可分离卷积优化平均池化
【代码】深度可分离卷积优化平均池化。原创 2024-09-23 15:44:10 · 224 阅读 · 0 评论 -
Tensor做为索引
【代码】Tensor做为索引。原创 2024-10-29 16:23:53 · 610 阅读 · 0 评论