`elasticai.creator.nn.fixed_point.linear.layer.batch_normed_linear`#

Module Contents#

Classes#

BatchNormedLinear

API#

class elasticai.creator.nn.fixed_point.linear.layer.batch_normed_linear.BatchNormedLinear(total_bits: int, frac_bits: int, in_features: int, out_features: int, bias: bool = True, bn_eps: float = 1e-05, bn_momentum: float = 0.1, bn_affine: bool = True, device: Any = None)[source]#

Bases: elasticai.creator.nn.design_creator_module.DesignCreatorModule, torch.nn.Module

property lin_weight: torch.Tensor#

property lin_bias: torch.Tensor | None#

property bn_weight: torch.Tensor#

property bn_bias: torch.Tensor#

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

create_design(name: str) → elasticai.creator.nn.fixed_point.linear.design.LinearDesign[source]#