src.gridmind.feature_construction.embedding_feature_extractor
=============================================================

.. py:module:: src.gridmind.feature_construction.embedding_feature_extractor


Classes
-------

.. autoapisummary::

   src.gridmind.feature_construction.embedding_feature_extractor.EmbeddingFeatureExtractor


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

.. py:class:: EmbeddingFeatureExtractor(embedding: torch.nn.Embedding)

   .. py:attribute:: embedding


   .. py:attribute:: device


   .. py:method:: __call__(input_index: int, *args: Any, **kwds: Any) -> Any