Rate this Page

LinearBn1d#

class torch.ao.nn.intrinsic.modules.fused.LinearBn1d(linear, bn)[source]#

This is a sequential container which calls the Linear and BatchNorm1d modules. During quantization this will be replaced with the corresponding fused module.