from ....Libs.Embedder.Classes.NanoSocratesEmbedder import NanoSocratesEmbedder import torch class BatchEmbedder(torch.nn.Module): def __init__(self, vocabulary_size: int, embedding_size: int) -> None: super().__init__() self.__embedder = NanoSocratesEmbedder(vocabulary_size,embedding_size) def forward(self, )