Rate this Page

BNReLU2d#

class torch.ao.nn.intrinsic.BNReLU2d(batch_norm, relu)[source]#

This is a sequential container which calls the BatchNorm 2d and ReLU modules. During quantization this will be replaced with the corresponding fused module.