Model Arch

This section is mainly intended for users who need to modify the model architecture. For most users, this part can be safely ignored.

Here, we provide a detailed introduction to several components of the model. Only the components recommended for users will be presented; components that are still under development will not be covered.

Contents

• Node Embedding
  • LinearNodeEmbedding
  • NonLinearNodeEmbedding
  • TensorNodeEmbedding
  • SO2TensorNodeEmbedding
• Edge Embedding
  • IdentityEdgeEmbedding
  • LinearEdgeEmbedding
  • NonLinearEdgeEmbedding
  • ElementEdgeEmbedding
• Edge Update
  • IdentityEdgeUpdate
  • ElementEdgeUpdate
  • Element2EdgeUpdate
• Interaction
  • CgtpInteraction
  • SO2Interaction
• Product
  • CgtpACE
  • GtpACE