ReductionLayer

class marius.nn.layers.ReductionLayer

__init__()

forward(self: marius._nn.layers.ReductionLayer, inputs: List[torch.Tensor]) → torch.Tensor

class marius.nn.layers.ConcatReduction

__init__(self: marius._nn.layers.ConcatReduction, layer_config: marius._config.LayerConfig, device: torch.device) → None

__init__(self: marius._nn.layers.ConcatReduction, input_dim: int, output_dim: int, device: Optional[torch.device] = None, init: marius._config.InitConfig, bias: bool = False, bias_init: marius._config.InitConfig, activation: str = ‘none’) → None

forward(self: marius._nn.layers.ConcatReduction, inputs: List[torch.Tensor]) → torch.Tensor

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

class marius.nn.layers.LinearReduction

__init__(self: marius._nn.layers.LinearReduction, layer_config: marius._config.LayerConfig, device: torch.device) → None

__init__(self: marius._nn.layers.LinearReduction, input_dim: int, output_dim: int, device: Optional[torch.device] = None, init: marius._config.InitConfig, bias: bool = False, bias_init: marius._config.InitConfig, activation: str = ‘none’) → None

forward(self: marius._nn.layers.LinearReduction, inputs: List[torch.Tensor]) → torch.Tensor

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

property weight_matrix