embeddings

Module: embeddings.py

This module implements embedding layers for Neural Networks

Authors

• Lokesh Mohanty (lokeshm@iisc.ac.in)

Version Info

• 06/01/2025: Initial version

Embedding

Bases: Embedding

Implements an Embedding layer

Source code in scirex/core/dl/nn/layers/embeddings.py

class Embedding(eqx.nn.Embedding):
    """
    Implements an Embedding layer
    """