Embeddings
----------

.. autoclass:: tn4ml.embeddings.Embedding
   :noindex:

Trigonometric
^^^^^^^^^^^^^
.. autoclass:: tn4ml.embeddings.trigonometric
   :members:
   :undoc-members:
   :noindex:

Fourier
^^^^^^^
.. autoclass:: tn4ml.embeddings.fourier
   :members:
   :undoc-members:
   :noindex:

Linear Complement
^^^^^^^^^^^^^^^^^^
.. autoclass:: tn4ml.embeddings.linear_complement_map
   :members:
   :undoc-members:

Gaussian RBF
^^^^^^^^^^^^
.. autoclass:: tn4ml.embeddings.gaussian_rbf
   :members:
   :undoc-members:

Polynomial
^^^^^^^^^^
.. autoclass:: tn4ml.embeddings.polynomial
   :members:
   :undoc-members:

Jax Arrays
^^^^^^^^^^
.. autoclass:: tn4ml.embeddings.jax_arrays
   :members:
   :undoc-members:

Patch Embedding
^^^^^^^^^^^^^^^
.. autoclass:: tn4ml.embeddings.PatchEmbedding
   :members:
   :undoc-members:

Embed Function
^^^^^^^^^^^^^^
.. autofunction:: tn4ml.embeddings.embed