Pytorch API
===========

.. py:currentmodule:: geometric_algebra_attention.pytorch

.. automodule:: geometric_algebra_attention.pytorch

Normalization layers
--------------------

.. autoclass:: MomentumNormalization

.. autoclass:: MomentumLayerNormalization

Vector-input functions
----------------------

.. autoclass:: VectorAttention
   :members:

.. autoclass:: Vector2Multivector
   :members:

.. autoclass:: Vector2VectorAttention
   :members:

.. autoclass:: LabeledVectorAttention
   :members:

.. autoclass:: TiedVectorAttention
   :members:

Multivector-input functions
---------------------------

.. autoclass:: MultivectorAttention
   :members:

.. autoclass:: Multivector2MultivectorAttention
   :members:

.. autoclass:: Multivector2Vector
   :members:

.. autoclass:: LabeledMultivectorAttention
   :members:

.. autoclass:: TiedMultivectorAttention
   :members: