評價此頁

BNReLU2d#

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

這是一個順序容器,它會呼叫 BatchNorm 2d 和 ReLU 模組。在量化過程中,它將被替換為相應的融合模組。