pytorch / 2 / generated / torch.ao.nn.intrinsic.bnrelu2d.html

BNReLU2d

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

This is a sequential container which calls the BatchNorm 2d and ReLU modules. During quantization this will be replaced with the corresponding fused module.

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.ao.nn.intrinsic.BNReLU2d.html