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

ConvBn2d

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

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