CVPR 2021 | Involution:超越convolution和self-attention的神经网络算子


论文地址:https://arxiv.org/pdf/2103.06255v2.pdf
代码地址:https://github.com/d-li14/involution
Involution
卷积,文章描述说它比convolution
更轻量更高效,形式上比self-attention
更加简洁,可以在各种视觉任务的模型上取得精度和效率的双重提升。
内卷积代码
import torch
from torch import nn