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

ConvReLU2d

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

This is a sequential container which calls the Conv2d 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.ConvReLU2d.html