src.gridmind.feature_construction.embedding_feature_extractor

Classes

EmbeddingFeatureExtractor

Module Contents

class src.gridmind.feature_construction.embedding_feature_extractor.EmbeddingFeatureExtractor(embedding: torch.nn.Embedding)[source]
embedding[source]
device[source]
__call__(input_index: int, *args: Any, **kwds: Any) Any[source]