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

ConvBn3d

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

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