mir-group/allegro

Question about the Tensor Product module.

floatingCatty opened this issue · 0 comments

Hello, I have been reading the code of allegro and trying to implement an E3-equivariant model recently. I am confused about the tensor product construction process in the allegro model. For example, in nn/_strided/_linear, and nn/_stride/_contract, some modules seem to realize the module to build tensor-production and linear module in e3nn. Still, when I compare the code with the e3nn module, I find many differences, such as the normalization, and some parameters such as the sparse mode.

So my question is, what are the functional differences comparing the tensor production module in allegro/nn concerning e3nn?