Jul, 2022

e3nn:欧几里德神经网络

TL;DRe3nn 是一种通用的框架,用于创建 E (3) 可变换的可训练函数,操作对象为描述三维系统的几何和几何张量,并能运用核心的张量积类和球谐函数来构建复杂的模块,例如卷积和注意力机制,可用于高效地处理张量场网络,三维中的可定向卷积神经网络,Clebsch-Gordan Networks,SE (3) 变换和其他 E (3) 可变换网络。