`elasticai.creator.base_modules.linear`#

Module Contents#

Classes#

`MathOperations`
`Linear`

API#

class elasticai.creator.base_modules.linear.MathOperations[source]#: Bases: elasticai.creator.base_modules.math_operations.Quantize, elasticai.creator.base_modules.math_operations.Add, elasticai.creator.base_modules.math_operations.MatMul, typing.Protocol

class elasticai.creator.base_modules.linear.Linear(in_features: int, out_features: int, operations: elasticai.creator.base_modules.linear.MathOperations, bias: bool, device: Any = None, dtype: Any = None)[source]#

Bases: torch.nn.Linear

forward(x: torch.Tensor) → torch.Tensor[source]#