BNReLU2d

classtorch.ao.nn.intrinsic.quantized.BNReLU2d(num_features, eps=1e-05, momentum=0.1, device=None, dtype=None)[源代码]

BNReLU2d 模块是将 BatchNorm2d 和 ReLU 融合在一起的模块

我们采用了与torch.ao.nn.quantized.BatchNorm2d相同的接口。

变量

torch.ao.nn.quantized.BatchNorm2d (与... 相同) –