3 Further, to consider multiple types of elementwise interaction, we use a set of M weight matrices per dimension.
for each Implementation-wise, we vertically stack G for M times to constructG ∈ R M d×3 , and use each row g i as input to Eq.