BNReLU2d

class torch.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 (同) –