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

ConvBnReLU2d

class torch.ao.nn.intrinsic.ConvBnReLU2d(conv, bn, relu) [source]

This is a sequential container which calls the Conv 2d, Batch Norm 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.ConvBnReLU2d.html