:py:mod:`ontolearn.nces_modules`
================================

.. py:module:: ontolearn.nces_modules

.. autoapi-nested-parse::

   NCES modules.



Module Contents
---------------

Classes
~~~~~~~

.. autoapisummary::

   ontolearn.nces_modules.MAB
   ontolearn.nces_modules.SAB
   ontolearn.nces_modules.ISAB
   ontolearn.nces_modules.PMA




.. py:class:: MAB(dim_Q, dim_K, dim_V, num_heads, ln=False)


   Bases: :py:obj:`torch.nn.Module`

   MAB module.

   .. py:method:: forward(Q, K)



.. py:class:: SAB(dim_in, dim_out, num_heads, ln=False)


   Bases: :py:obj:`torch.nn.Module`

   SAB module.

   .. py:method:: forward(X)



.. py:class:: ISAB(dim_in, dim_out, num_heads, num_inds, ln=False)


   Bases: :py:obj:`torch.nn.Module`

   ISAB module.

   .. py:method:: forward(X)



.. py:class:: PMA(dim, num_heads, num_seeds, ln=False)


   Bases: :py:obj:`torch.nn.Module`

   PMA module.

   .. py:method:: forward(X)