技术栈

conv2d

Briwisdom
2 小时前
gemm·fft·matmul·conv2d·implicit gemm·winograd
Conv2D 算子实现深入——im2col、Implicit GEMM、Winograd、FFT 代码级拆解副标题: 一个卷积算子,四种实现路线。本文不再停留在概念层面,而是用可运行的代码逐一拆解:直接卷积怎么算、im2col 怎么展开成矩阵、Implicit GEMM 怎么在 kernel 内部在线计算访存地址、Winograd F(2,3) 的三组变换矩阵从哪来、FFT 怎么利用卷积定理做一次频域点乘。训练 vs 推理的差异分析见 博客 #34。
闻道且行之
5 个月前
人工智能·pytorch·python·深度学习·conv2d
Pytorch之torch.nn.Conv2d详解卷积层是卷积神经网络(CNN)的核心组件,而torch.nn.Conv2d作为 PyTorch 中实现二维卷积的核心类,广泛应用于图像分类、目标检测、语义分割等计算机视觉任务。本文将从基础概念、参数详解、使用示例到核心原理,全方位拆解torch.nn.Conv2d,帮助新手快速掌握其用法和底层逻辑。
我是有底线的