gatr_outer_product

cuequivariance.descriptors.gatr_outer_product() → SegmentedTensorProduct

subsrcipts: input1[i],input2[j],output[k]+ijk

References

• https://arxiv.org/pdf/2305.18415

• Source code

Examples

>>> d = gatr_outer_product(); d
i,j,k+ijk sizes=16,16,16 num_segments=8,8,8 num_paths=30 i={1, 3} j={1, 3} k={1, 3}