napistu\_torch.models.heads
===========================

.. automodule:: napistu_torch.models.heads

   
   .. rubric:: Classes

   .. autosummary::
   
      AttentionHead
      ConditionalRotatEHead
      Decoder
      DistMultHead
      DotProductHead
      EdgeMLPHead
      NodeClassificationHead
      RelationAttentionHead
      RelationAttentionMLPHead
      RelationGatedMLPHead
      RotatEHead
      TransEHead