bitorch.models.dlrm.create_embeddings

bitorch.models.dlrm.create_embeddings(embedding_dimension: int, layer_sizes: List[int], quantized: bool, sparse: bool = False) → ModuleList

creates the embedding layers for each category.