ConvRelu2d¶

class ConvRelu2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, conv_mode='cross_correlation', compute_mode='default', padding_mode='zeros', **kwargs)[源代码]¶: 融合了 Conv2d 和 relu 的 Module. 可以调用 quantize_qat 替换成 QATModule 型的 ConvRelu2d.

PixelShuffle

ConvBn2d