Jan, 2024

conv_einsum:卷积张量神经网络中多线性操作的表示和快速计算框架

TL;DR这篇论文提出了一种用于表示张量卷积层的统一框架,以及一种能够以最小化浮点操作数(FLOPs)的方式评估这些框架的方法,实验证明 conv_einsum 显著提高了卷积张量网络的计算和内存效率。