FeatureLayer

class marius.nn.layers.FeatureLayer

__init__(self: marius._nn.layers.FeatureLayer, layer_config: marius._config.LayerConfig, device: torch.device, offset: int = 0) → None

__init__(self: marius._nn.layers.FeatureLayer, dimension: int, device: torch.device, bias: bool = False, bias_init: marius._config.InitConfig, activation: str = ‘none’, offset: int = 0) → None

forward(self: marius._nn.layers.EmbeddingLayer, input: torch.Tensor) → torch.Tensor

property offset

reset(self: marius._nn.layers.FeatureLayer) → None