BNReLU2d¶
- class torch.ao.nn.intrinsic.quantized.BNReLU2d(num_features, eps=1e-05, momentum=0.1, device=None, dtype=None)[source][source]¶
A BNReLU2d module is a fused module of BatchNorm2d and ReLU
We adopt the same interface as
torch.ao.nn.quantized.BatchNorm2d
.- Variables
torch.ao.nn.quantized.BatchNorm2d (Same as) –