bitorch_engine.layers.qembedding.binary.layer

Classes

BinaryEmbeddingBag(*args, num_embeddings, ...)	An binary embedding bag implementation.
BinaryEmbeddingBagForward(*args, **kwargs)	An experimental PyTorch function for forward pass of binary embedding bag.
BinaryEmbeddingCuda(*args, num_embeddings, ...)	Binarized version of embedding layer, currently in experimental stage.
BinaryEmbeddingForward(*args, **kwargs)	Experimental class for forward pass of binary embeddings.
BinaryEmbeddingParameter([data, ...])	A custom parameter class for binary embeddings, extending torch.nn.Parameter.