BNReLU3d¶

class torch.ao.nn.intrinsic.quantized.BNReLU3d(num_features, eps=1e-05, momentum=0.1, device=None, dtype=None)[source][source]¶

BNReLU3d 模块是 BatchNorm3d 和 ReLU 的融合模块

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